Implement numpy based prediction feature

2025-03-16 09:47:11 -04:00
parent 884c173c0c
commit 12408c33f4
5 changed files with 402 additions and 31 deletions
--- a/src/toldg/models.py
+++ b/src/toldg/models.py
@@ -50,6 +50,7 @@ class Config(BaseModel):
    output_file: Path = Path("output.ldg")
    csv_configs: List[CsvConfig]
    categories: List[str]
+    model: Path = Path("transaction_classifier.pkl")


 class Mapping(BaseModel):