Skip to content

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

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.