configs/model_config.yaml

# MLP模型配置文件
# MLP Model Configuration

# 模型基本信息
model_info:
  name: "MLP_Emotion_Predictor"
  type: "MLP"
  version: "1.0"

# 输入输出维度
dimensions:
  input_dim: 7    # 输入维度：User PAD 3维 + Vitality 1维 + Current PAD 3维
  output_dim: 3   # 输出维度：ΔPAD 3维（压力通过 PAD 变化动态计算）

# 网络架构参数
architecture:
  # 隐藏层配置
  hidden_layers:
    - size: 512
      activation: "ReLU"
      dropout: 0.05
    - size: 256
      activation: "ReLU"
      dropout: 0.05
    - size: 128
      activation: "ReLU"
      dropout: 0.05
  
  # 输出层配置
  output_layer:
    activation: "Linear"  # 线性激活，用于回归任务
  
  # 批归一化
  use_batch_norm: false
  
  # 层归一化
  use_layer_norm: false

# 初始化参数
initialization:
  weight_init: "xavier_uniform"  # xavier_uniform, xavier_normal, kaiming_uniform, kaiming_normal
  bias_init: "zeros"

# 正则化参数
regularization:
  # L2正则化
  weight_decay: 0
  
  # Dropout
  dropout_config:
    type: "standard"  # standard, variational
    rate: 0.2

# 模型保存配置
model_saving:
  save_best_only: true
  save_format: "pytorch"  # pytorch, onnx, torchscript
  checkpoint_interval: 10  # 每10个epoch保存一次检查点

# 模型特定配置
emotion_model:
  # PAD情绪空间的特殊配置
  pad_space:
    # PAD值的范围限制
    pleasure_range: [-1.0, 1.0]     # 快乐维度
    arousal_range: [-1.0, 1.0]      # 激活度维度  
    dominance_range: [-1.0, 1.0]    # 支配度维度
  
  # 生理指标配置
  vitality:
    range: [0.0, 1.0]               # 活力值范围
    normalization: "min_max"        # min_max, z_score, robust
  
  # 预测输出配置
  prediction:
    # ΔPAD的变化范围限制
    delta_pad_range: [-0.5, 0.5]    # PAD变化的合理范围
    # 压力值变化范围
    delta_pressure_range: [-0.3, 0.3]
    # 置信度范围
    confidence_range: [0.0, 1.0]