PocketAccountant / src /finetune /__init__.py
eldinosaur's picture
English-default UI + bilingual agent + USA tax support (Schedule C/SE/federal) and expanded US+EN regulation corpus
56ed47e verified
Raw
History Blame Contribute Delete
880 Bytes
"""Fine-tune — SAT transaction classifier (🎯 Well-Tuned).
The catalog is the single source of truth: it drives both the training dataset and
the deterministic fallback classifier, so they never disagree on a label. The
fine-tuned MiniCPM generalizes beyond the keyword list (messy OCR, unseen brands,
Spanish variation); the rule classifier is the honest fallback until weights load.
"""
from .catalog import ALL_CATEGORIES, Category
from .classifier import (
Classification,
ModelClassifier,
RemoteClassifier,
RuleClassifier,
build_prompt,
parse_label,
)
from .dataset import build_examples, split, write_jsonl
__all__ = [
"Category",
"ALL_CATEGORIES",
"Classification",
"RuleClassifier",
"ModelClassifier",
"RemoteClassifier",
"build_prompt",
"parse_label",
"build_examples",
"split",
"write_jsonl",
]