This commit is contained in:
2509165008
2026-04-30 16:05:45 +08:00
parent 20993e9fae
commit 66621ab351
36 changed files with 7771 additions and 3 deletions

View File

@@ -20,15 +20,15 @@ KEEP_PROB = 1.0 # Dropout保留概率LR忽略
# ==================== 训练相关 ====================
LEARNING_RATE = 0.05 # 学习率
NUM_EPOCHS = 100 # 训练轮数
NUM_EPOCHS = 60 # 训练轮数
BATCH_SIZE = 64 # 批次大小
# ==================== 类别权重(解决数据不平衡问题)====================
USE_CLASS_WEIGHT = True # True=启用类别权重, False=不启用(对比用)
# 权重计算公式: n_samples / (n_classes * n_class_i)
# 正面评论多所以权重小,负面评论少所以权重大
CLASS_WEIGHT_POS = 0.73 # 正面类权重(自动计算)
CLASS_WEIGHT_NEG = 1.58 # 负面类权重(自动计算)
CLASS_WEIGHT_POS = 0.73 # 正面类权重(自动计算)
CLASS_WEIGHT_NEG = 2.00 # 负面类权重(自动计算)
# ==================== 实验相关 ====================
RUN_COMPARISON = False # True=运行对比实验, False=运行单个模型