训练完keras模型后怎么画出loss和val_loss的对比图?

  统计/机器学习 模型验证 数据可视化 人工神经网络 TensorFlow    浏览次数:206        分享
0

我用keras训练了一个模型

model.fit(train_array, labels, batch_size=32, epochs=10, validation_split=0.3)

返回了如下的loss和val_loss

Epoch 1/10
17500/17500 [==============================] - 2s 120us/step - loss: 0.6638 - val_loss: 0.6066
Epoch 2/10
17500/17500 [==============================] - 1s 62us/step - loss: 0.5379 - val_loss: 0.4775
Epoch 3/10
17500/17500 [==============================] - 1s 61us/step - loss: 0.4247 - val_loss: 0.3939
Epoch 4/10
17500/17500 [==============================] - 1s 63us/step - loss: 0.3525 - val_loss: 0.3480
Epoch 5/10
17500/17500 [==============================] - 1s 62us/step - loss: 0.3077 - val_loss: 0.3162
Epoch 6/10
17500/17500 [==============================] - 1s 61us/step - loss: 0.2783 - val_loss: 0.2991
Epoch 7/10
17500/17500 [==============================] - 1s 61us/step - loss: 0.2573 - val_loss: 0.2878
Epoch 8/10
17500/17500 [==============================] - 1s 60us/step - loss: 0.2418 - val_loss: 0.2815
Epoch 9/10
17500/17500 [==============================] - 1s 63us/step - loss: 0.2299 - val_loss: 0.2779
Epoch 10/10
17500/17500 [==============================] - 1s 62us/step - loss: 0.2203 - val_loss: 0.2750

我怎么根据这个loss和val_loss画图?就像底下这个图的效果


 

xkk1o   2019-07-18 08:06



   1个回答 
3

你要把model.fit的训练过程保存下来

history = model.fit(...)

history是一个dict,dict里有history子dict,里面有loss和val_loss,都是list的形式,然后就可以正常画图了

epochs = len(history.history['loss'])
plt.plot(range(epochs), history.history['loss'], label='loss')
plt.plot(range(epochs), history.history['val_loss'], label='val_loss')
plt.legend()
plt.show()
SofaSofa数据科学社区DS面试题库 DS面经

abuu   2019-07-21 23:12

很有帮助,谢谢! - xkk1o   2019-07-27 14:20


  相关主题

为什么Keras和Pytorch都是把模型保存成h5的格式而不是pkl格式?   1回答

怎么在keras中加载MNSIT数据集?   2回答

keras load_model报错   1回答

keras怎么设置/改动rmsprop的学习率?   1回答

keras里的predict,predict_classes,predict_proba有什么区别?   2回答

keras sequential model怎么增加一个dropout层?   1回答

keras加载imdb数据时num_words参数是干嘛的?   1回答

keras.models.Sequential()是一种什么模型   1回答

keras hdf5 從某層輸入數值   1回答

keras模型合并请教   2回答

keras中怎么增加Leaky relu的激活层   2回答

keras里layer默认的激活函数是什么?   1回答



回答问题时需要注意什么?

我们谢绝在回答前讲“生动”的故事。

我们谢绝“这么简单,你自己想”、“书上有的,你认真看”这类的回答;如果你认为对方的提问方式或者内容不妥,你可以直接忽略该问题,不用进行任何作答,甚至可以对该问题投反对票。

我们谢绝答非所问。

我们谢绝自己不会、硬要回答。

我们感激每一个用户在编写答案时的努力与付出!