如何使用Tensorflow来测试、重置模型并加载最新的检查点?


Tensorflow 是 Google 提供的一个机器学习框架。它是一个开源框架,与 Python 结合使用,用于实现算法、深度学习应用程序等等。它用于研究和生产目的。它具有优化技术,有助于快速执行复杂的数学运算。这是因为它使用了 NumPy 和多维数组。这些多维数组也称为“张量”。该框架支持使用深度神经网络。它具有高度可扩展性,并附带许多流行的数据集。

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

pip install tensorflow

Keras 是作为 ONEIROS(开放式神经电子智能机器人操作系统)项目研究的一部分开发的。Keras 是一个深度学习 API,是用 Python 编写的。它是一个高级 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 previously saved weights are loaded")
model.load_weights(latest)
print("The model is being re−evaluated")
loss, acc = model.evaluate(test_images, test_labels, verbose=2)
print("This is the restored model, with accuracy: {:5.3f}%".format(100 * acc))

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

输出

A new model instance is created
The previously saved weights are loaded
The model is being re-evaluated
32/32 - 0s - loss: 0.4828 - sparse_categorical_accuracy: 0.8770
This is the restored model, with accuracy:87.700%

解释

  • 再次,使用“create_model”方法创建实例的新模型。

  • 使用“load_weights”方法将先前保存的权重加载到此实例。

  • 使用“evaluate”方法评估此新模型。

  • 确定其在训练期间的准确性和损失。

  • 这些值显示在控制台上。

更新于: 2021年1月20日

179 次查看

开启您的 职业生涯

通过完成课程获得认证

立即开始
广告