如下所示:
1
2
3
4
5
|
H = model.fit(trainX, trainY, validation_data=(testX, testY),batch_size=BS, epochs=EPOCHS, verbose=1)
plt.style.use("ggplot")
plt.figure()
plt.plot(np.arange(0, EPOCHS), H.history["loss"], label="train_loss")
|
History类对象包含两个属性,分别为epoch和history,epoch为训练轮数。history字典类型,包含val_loss,val_acc,loss,acc四个key值。
History.history属性会把之前的那些值全保存在里面,如果有验证集的话,也包含了验证集的这些指标变化情况
补充知识:
原文链接:https://blog.csdn.net/weixin_33750452/article/details/89623748