modularml.core.model_graph.model_graph

DAG of FeatureSets and ModelStages

Responsibilities:

Stores all nodes: FeatureSets and ModelStages. Stores all edges: how outputs of one stage feed another. Performs:

Topological sorting Lazy build() propagation of shapes Forward pass through the graph

Functions

make_dummy_batch(feature_shape[, batch_size])
make_dummy_data(shape[, batch_size])	Creates a dummy Data object.

Classes

ModelGraph(nodes)

modularml.core.model_graph.model_graph.make_dummy_data(shape: Tuple[int, ...], batch_size: int = None) → Data

Creates a dummy Data object. A batch dimension is added if batch_size is not None.