keras 保存模型导入模型，保存训练记录，导入训练记录，绘制训练曲线

使用keras保存导入模型¹

from keras.models import load_model

model.save('my_model.h5')  # creates a HDF5 file 'my_model.h5'
del model  # deletes the existing model

# returns a compiled model
# identical to the previous one
model = load_model('my_model.h5')

使用keras保存训练记录

记录训练记录

hist = model.fit(x, y, validation_split=0.2)
print(hist.history)

保存到硬盘

hist = model.fit(x, y, validation_split=0.2)
print(hist.history)

返回的hist对象中包含训练历史输出，即在terminal中输出的数据，用于绘制训练曲线

使用keras绘制训练曲线

下面是绘制准确率的变化曲线，如果将acc修改成loss就可以绘制损失变化曲线，对于其他metrics是相同的道理

plt.plot(history.history['acc'])
plt.plot(history.history['val_acc'])
plt.title("model accuracy")
plt.ylabel("Accuracy")
plt.xlabel("epoch")
plt.legend(["train","test"],loc="lower right")
plt.show()

---

1. Keras ↩︎