提交 c2dee93b 编写于 作者: 片刻小哥哥's avatar 片刻小哥哥

优化学习率,效果更加平稳

上级 9f8a2b68
......@@ -15,7 +15,7 @@ class Config(object):
# 根据前六个字预测第七个字
max_len = 6
batch_size = 512
learning_rate = 0.005
learning_rate = 0.0005
pre_num = 3
MAX_SEQUENCE_LENGTH = 1000 # 每个文本或者句子的截断长度,只保留1000个单词
EMBEDDING_DIM = 60 # 词向量维度
......@@ -163,7 +163,7 @@ class EmotionModel(object):
preds = Dense(self.pre_num, activation='softmax')(x)
self.model = Model(sequence_input, preds)
# 设置优化器
optimizer = Adam(lr=self.config.learning_rate)
optimizer = Adam(lr=self.config.learning_rate, beta_1=0.95, beta_2=0.999,epsilon=1e-08)
self.model.compile(optimizer=optimizer, loss='categorical_crossentropy', metrics=['accuracy'])
self.model.summary()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册