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")