如何在Python中使用Keras以hdf5格式保存模型?


TensorFlow是由Google提供的机器学习框架。它是一个开源框架,与Python结合使用,可以实现算法、深度学习应用程序等等。它用于研究和生产目的。

可以使用以下代码行在Windows上安装“tensorflow”包:

pip install tensorflow

张量是TensorFlow中使用的数据结构。它有助于连接数据流图中的边。这个数据流图被称为“数据流图”。张量只不过是多维数组或列表。

Keras是ONEIROS(开放式神经电子智能机器人操作系统)项目研究的一部分而开发的。Keras是一个用Python编写的深度学习API。它是一个高级API,具有高效的界面,有助于解决机器学习问题。

它运行在TensorFlow框架之上。它是为了帮助快速实验而构建的。它提供了开发和封装机器学习解决方案所必需的基本抽象和构建块。它具有高度的可扩展性,并具有跨平台能力。这意味着Keras可以在TPU或GPU集群上运行。Keras模型也可以导出到Web浏览器或移动电话中运行。

Keras已经存在于TensorFlow包中。可以使用以下代码行访问它。

import tensorflow
from tensorflow import keras

我们使用Google Colaboratory运行以下代码。Google Colab或Colaboratory有助于通过浏览器运行Python代码,无需任何配置,并可以免费访问GPU(图形处理单元)。Colaboratory构建在Jupyter Notebook之上。以下是代码:

示例

print("The model is saved to HDF5 format")
model.save('my_model.h5')
print("The same model is recreated with same weights and optimizer")
new_model = tf.keras.models.load_model('my_model.h5')
print("The architecture of the model is observed")
new_model.summary()

代码来源: https://tensorflowcn.cn/tutorials/keras/save_and_load

输出

解释

  • 可以使用“save”函数保存新创建的模型。

  • 可以使用扩展名“h5”将其专门保存为hdf5格式。

  • 此模型使用之前的权重和优化器加载。

  • 使用“summary”方法在控制台上显示有关新模型的详细信息。

更新于:2021年1月21日

318 次查看

启动你的职业生涯

完成课程获得认证

开始学习
广告