machine-learning - 如何保存 keras 实验运行产生的数据结果？

Question

我想将实验结果保存在 keras 而不是模型中。例如，我想保存以下结果的所有内容:

''' Plots '''
if plot:
    # Plots for training and testing process: loss and accuracy
    plt.figure(0)
    plt.plot(cnn.history['acc'],'r')
    plt.plot(cnn.history['val_acc'],'g')
    plt.xticks(np.arange(0, nb_epochs+1, 2.0))
    plt.rcParams['figure.figsize'] = (8, 6)
    plt.xlabel("Num of Epochs")
    plt.ylabel("Accuracy")
    plt.title("Training Accuracy vs Validation Accuracy")
    plt.legend(['train','validation'])


    plt.figure(1)
    plt.plot(cnn.history['loss'],'r')
    plt.plot(cnn.history['val_loss'],'g')
    plt.xticks(np.arange(0, nb_epochs+1, 2.0))
    plt.rcParams['figure.figsize'] = (8, 6)
    plt.xlabel("Num of Epochs")
    plt.ylabel("Loss")
    plt.title("Training Loss vs Validation Loss")
    plt.legend(['train','validation'])

如何保存所有这些，以便我可以再次绘制图并检查训练期间发生的情况？

网站:

https://keras.io/getting-started/faq/#how-can-i-save-a-keras-model

似乎没有解释...帮助？

Answer 1

pickle 模块可以让你序列化 python 对象。

您可以使用以下方式保存历史记录:

pkl.dump(cnn.history, file_obj)

如果您想将绘图保存为图像:

plt.savefig(path)

您还可以尝试pickle matplotlib Figure/Axes 对象来重新创建交互式绘图，但此功能是实验性的。我建议只腌制你的历史字典，然后用上面的代码重新生成绘图。