File size: 1,496 Bytes
0a6452f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 |
"""
模型模块
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情绪和生理状态变化预测模型" |