generated from felixm/defaultpy
14 lines
421 B
Python
14 lines
421 B
Python
import logging
|
|
|
|
from toldg.models import Config
|
|
from toldg.predict import train_classifier
|
|
from toldg.process import process_csv_files
|
|
|
|
|
|
def train(config: Config):
|
|
"""Train a transaction classifier from csv files."""
|
|
# Process transactions to get training data
|
|
transactions = process_csv_files(config)
|
|
train_classifier(transactions, config.categories, config.model)
|
|
logging.info("Training completed")
|