如何使用 Keras 和 Python 保存整个模型?
Tensorflow 是 Google 提供的一个机器学习框架。它是一个开源框架,与 Python 结合使用,用于实现算法、深度学习应用程序等等。它用于研究和生产目的。
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("A new model instance is created") model = create_model() print("The model is fit to the training data") model.fit(train_images, train_labels, epochs=5) print("The model is saved") !mkdir -p saved_model model.save('saved_model/my_model') ls saved_model
代码来源 - https://tensorflowcn.cn/tutorials/keras/save_and_load
输出
解释
使用“create_model”方法创建新的模型。
将此新模型拟合到训练数据。
创建一个新目录来存储新模型。
拟合完成后,使用“save”方法保存它。
保存模型的路径显示在控制台上。
广告