File size: 617 Bytes
94d2494
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
"""src.models – Model registry, training, and metrics."""

from src.models.registry import (
    detect_task_type,
    get_classification_configs,
    get_model_configs,
    get_regression_configs,
)
from src.models.metrics import compute_metrics, compute_regression_metrics
from src.models.training import auto_train, evaluate_all_models, train_single_model

__all__ = [
    "auto_train",
    "compute_metrics",
    "compute_regression_metrics",
    "detect_task_type",
    "evaluate_all_models",
    "get_classification_configs",
    "get_model_configs",
    "get_regression_configs",
    "train_single_model",
]