src/models/__init__.py

"""
模型模块
Model module for emotion and physiological state prediction

该模块包含了PAD预测器的所有核心组件：
- PADPredictor: 主要的预测模型
- 损失函数: 各种训练损失函数
- 评估指标: 模型评估和校准指标
- 模型工厂: 从配置创建模型和其他组件
"""

# 核心模型
from .pad_predictor import PADPredictor, create_pad_predictor

# 损失函数
from .loss_functions import (
    WeightedMSELoss,
    ConfidenceLoss,
    AdaptiveWeightedLoss,
    FocalLoss,
    MultiTaskLoss,
    create_loss_function
)

# 评估指标
from .metrics import (
    RegressionMetrics,
    CalibrationMetrics,
    PADMetrics,
    create_metrics
)

# 模型工厂
from .model_factory import (
    ModelFactory,
    model_factory,
    create_model_from_config,
    create_training_setup,
    save_model_config
)

__all__ = [
    # 核心模型
    "PADPredictor",
    "create_pad_predictor",
    
    # 损失函数
    "WeightedMSELoss",
    "ConfidenceLoss",
    "AdaptiveWeightedLoss",
    "FocalLoss",
    "MultiTaskLoss",
    "create_loss_function",
    
    # 评估指标
    "RegressionMetrics",
    "CalibrationMetrics",
    "PADMetrics",
    "create_metrics",
    
    # 模型工厂
    "ModelFactory",
    "model_factory",
    "create_model_from_config",
    "create_training_setup",
    "save_model_config",
]

# 版本信息
__version__ = "1.0.0"
__author__ = "PAD Predictor Team"
__description__ = "PAD情绪和生理状态变化预测模型"