File size: 408 Bytes
19ce9e8
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
"""
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']