archaeo_super_prompt.modeling.predict
source module archaeo_super_prompt.modeling.predict
Code containing the global model and a way to infer into it.
Functions
-
build_complete_inference_dag — Build the inference model from fitted parts.
-
score_dag — From an already fitted model, apply scoring over the extractors.
source build_complete_inference_dag(parts: ExtractionDAGParts)
Build the inference model from fitted parts.
source score_dag(parts: ExtractionDAGParts, inputs: PDFPathDataset, eval_ds: MagohDataset)
From an already fitted model, apply scoring over the extractors.