h5py文件和pickle文件保存模型的区别

Ind*_*Ind 5 python pickle python-3.x h5py keras

我想要获得模型保存的清晰图像。神经网络中有超参数和模型。

训练模型后,我想保存所有内容以使用它们,而无需重新训练模型。

当我将模型保存为 h5py 文件(.H5)时,它是否也会保存超参数?

如果是,pickle 文件的用途是什么?

sen*_*nce 8

来自how-i-save-a-keras-model

您可以将model.save(filepath)Keras 模型保存到单个 HDF5文件中,其中包含:

  • 模型的架构,允许重新创建模型
  • 模型的权重
  • 训练配置(损失、优化器)
  • 优化器的状态,允许从您上次中断的地方恢复训练。

它明确表示:

不建议使用 pickle 或 cPickle 保存 Keras 模型。

为什么?嗯,这可能有很多原因。是一个很好的起点。