diff --git a/src/py3.x/tensorflow2.x/Emotion_acc.png b/src/py3.x/tensorflow2.x/Emotion_acc.png index a323c605..d40f1a2c 100644 Binary files a/src/py3.x/tensorflow2.x/Emotion_acc.png and b/src/py3.x/tensorflow2.x/Emotion_acc.png differ diff --git a/src/py3.x/tensorflow2.x/Emotion_loss.png b/src/py3.x/tensorflow2.x/Emotion_loss.png index 830d4b51..8bbb6cc9 100644 Binary files a/src/py3.x/tensorflow2.x/Emotion_loss.png and b/src/py3.x/tensorflow2.x/Emotion_loss.png differ diff --git a/src/py3.x/tensorflow2.x/config.py b/src/py3.x/tensorflow2.x/config.py index b865d28b..7a7322f0 100644 --- a/src/py3.x/tensorflow2.x/config.py +++ b/src/py3.x/tensorflow2.x/config.py @@ -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 # 词向量维度 diff --git a/src/py3.x/tensorflow2.x/text_Emotion.py b/src/py3.x/tensorflow2.x/text_Emotion.py index 3d1d5bf0..b8437a76 100644 --- a/src/py3.x/tensorflow2.x/text_Emotion.py +++ b/src/py3.x/tensorflow2.x/text_Emotion.py @@ -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()