完成作业3-2-2
This commit is contained in:
@@ -10,13 +10,13 @@
|
||||
DATA_DIR = 'data/ChnSentiCorp' # 数据集路径
|
||||
MAX_FEATURES = 3000 # 词表最大容量
|
||||
MAX_SEQ_LEN = 100 # 句子最大长度(词数)
|
||||
VECTORIZER_TYPE = 'tfidf' # 'tfidf' 或 'bow'(向量化方式)
|
||||
VECTORIZER_TYPE = 'bow' # 'tfidf' 或 'bow'(向量化方式)
|
||||
|
||||
# ==================== 模型相关 ====================
|
||||
MODEL_TYPE = 'mlp' # 'mlp' 或 'lr'(模型类型)
|
||||
HIDDEN_SIZE = 64 # MLP隐藏层大小(LR忽略)
|
||||
HIDDEN_SIZE = 128 # MLP隐藏层大小(LR忽略)
|
||||
NUM_CLASSES = 2 # 类别数(正面/负面二分类)
|
||||
KEEP_PROB = 1.0 # Dropout保留概率(LR忽略,设为1即可)
|
||||
KEEP_PROB = 0.8 # Dropout保留概率(LR忽略,设为1即可)
|
||||
|
||||
# ==================== 训练相关 ====================
|
||||
LEARNING_RATE = 0.05 # 学习率
|
||||
@@ -31,10 +31,11 @@ CLASS_WEIGHT_POS = 0.73 # 正面类权重(自动计算)
|
||||
CLASS_WEIGHT_NEG = 1.58 # 负面类权重(自动计算)
|
||||
|
||||
# ==================== 实验相关 ====================
|
||||
RUN_COMPARISON = False # True=运行对比实验, False=运行单个模型
|
||||
RUN_COMPARISON = True # True=运行对比实验, False=运行单个模型
|
||||
COMPARE_MODELS = ['lr', 'mlp'] # 要对比的模型列表
|
||||
COMPARE_VECTORS = ['bow', 'tfidf'] # 要对比的向量化方式
|
||||
|
||||
# ==================== 其他 ====================
|
||||
RANDOM_SEED = 42 # 随机种子(保证可复现)
|
||||
VERBOSE = True # 打印详细日志
|
||||
|
||||
|
||||
Reference in New Issue
Block a user