keras History对象的用法

时间:2021-12-03 18:52:03

如下所示:

?
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