thadillo
Add fine-tuning infrastructure with custom hyperparameters and dataset import
19ce9e8
raw
history blame contribute delete
408 Bytes
"""
Fine-tuning module for training custom classification models.
This module provides tools for:
- Preparing training datasets from admin corrections
- Fine-tuning BART models using LoRA (Low-Rank Adaptation)
- Evaluating model performance
- Managing model versions and deployment
"""
from .trainer import BARTFineTuner
from .model_manager import ModelManager
__all__ = ['BARTFineTuner', 'ModelManager']