Implement numpy based prediction feature

This commit is contained in:
2025-03-16 09:47:11 -04:00
parent 884c173c0c
commit 12408c33f4
5 changed files with 402 additions and 31 deletions

View File

@@ -1,9 +1,14 @@
from toldg.models import Config, CsvConfig, Mapping, Transaction
import logging
from pathlib import Path
from toldg.models import Config
from toldg.predict import train_classifier
from toldg.process import process_csv_files
def train(config: Config):
print("[train] start")
"""Train a transaction classifier from csv files."""
# Process transactions to get training data
transactions = process_csv_files(config)
for t in transactions:
pass
train_classifier(transactions, config.categories, config.model)
logging.info("Training completed")