完成作业
This commit is contained in:
@@ -13,13 +13,13 @@ MAX_SEQ_LEN = 100 # 句子最大长度(词数)
|
||||
VECTORIZER_TYPE = 'tfidf' # 'tfidf' 或 'bow'(向量化方式)
|
||||
|
||||
# ==================== 模型相关 ====================
|
||||
MODEL_TYPE = 'mlp' # 'mlp' 或 'lr'(模型类型)
|
||||
MODEL_TYPE = 'lr' # 'mlp' 或 'lr'(模型类型)
|
||||
HIDDEN_SIZE = 64 # MLP隐藏层大小(LR忽略)
|
||||
NUM_CLASSES = 2 # 类别数(正面/负面二分类)
|
||||
KEEP_PROB = 1.0 # Dropout保留概率(LR忽略,设为1即可)
|
||||
KEEP_PROB = 1.0 # Dropout保留概——0.06率(LR忽略,设为1即可)
|
||||
|
||||
# ==================== 训练相关 ====================
|
||||
LEARNING_RATE = 0.05 # 学习率
|
||||
LEARNING_RATE = 0.08 # 学习率
|
||||
NUM_EPOCHS = 100 # 训练轮数
|
||||
BATCH_SIZE = 64 # 批次大小
|
||||
|
||||
|
||||
Reference in New Issue
Block a user