Skip to content

archaeo_super_prompt.modeling.DAG_builder

source module archaeo_super_prompt.modeling.DAG_builder

Interface over the DAG builder of skdag.

Classes

  • DAGComponent A component for the.

  • DAGBuilder A DAG builder suited for a handling with Transformers and Dataframes.

source class DAGComponent[Estimator: BaseEstimator]()

Bases : NamedTuple

A component for the.

source class DAGBuilder()

A DAG builder suited for a handling with Transformers and Dataframes.

Initiate an empty pipeline.

Methods

source method DAGBuilder.add_node(component: DAGComponent, deps: list[DAGComponent] = [])

Add a component to the DAG pipeline.

source method DAGBuilder.add_linearly_chained_nodes(components: list[DAGComponent], start_anchor: list[DAGComponent] = [])

Add a chain of components.

source method DAGBuilder.make_dag()

"Return the dag.