Spaces:
Sleeping
Sleeping
| """Training module""" | |
| from .metrics import ( | |
| SegmentationMetrics, | |
| ClassificationMetrics, | |
| MetricsTracker, | |
| EarlyStopping, | |
| get_metrics_tracker | |
| ) | |
| from .trainer import Trainer, get_trainer | |
| from .classifier import ForgeryClassifier, get_classifier | |
| __all__ = [ | |
| 'SegmentationMetrics', | |
| 'ClassificationMetrics', | |
| 'MetricsTracker', | |
| 'EarlyStopping', | |
| 'get_metrics_tracker', | |
| 'Trainer', | |
| 'get_trainer', | |
| 'ForgeryClassifier', | |
| 'get_classifier' | |
| ] | |