Skip to content

archaeo_super_prompt.modeling.other_dag

source module archaeo_super_prompt.modeling.other_dag

DAGs to train the FieldExtractor models.

Classes

  • ExtractionDAGParts A decomposition of the general DAG into different parts for a better handling between the training, the inference and the evaluation modes.

Functions

source class ExtractionDAGParts()

Bases : NamedTuple

A decomposition of the general DAG into different parts for a better handling between the training, the inference and the evaluation modes.

source get_advanced_pipeline()DAGBuilder

Return the most advanced pre-processing DAG for the model.

All its estimators and transformers are initialized with particular parametres.

Returns

  • DAGBuilder A part of the complete DAG for getting the pre-processed data. The field extractors related to their parent node, to apply on these extractors special training or evaluation operations or to bind them to the preprocessing dag The final union component to finish the building of the complete DAG in inference mode.