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属性会把之前的那些值全保存在里面,如果有验证集的话,也包含了验证集的这些指标变化情况