保存下载的数据和训练好的模型

Saving downloaded data as well as trained model

有没有办法将特定代码块的执行保存在笔记本中,这样我就不必再次 运行 了。重新加载后可以继续其余代码吗?
例如,

from __future__ import absolute_import, division, print_function, unicode_literals
import tensorflow as tf
from tensorflow.keras import datasets, layers, models
import matplotlib.pyplot as plt
import pandas as pd
import numpy as np

(train_images1, train_labels), (test_images1, test_labels) = datasets.cifar10.load_data()

# Normalize pixel values to be between 0 and 1
train_images, test_images = train_images1 / 255.0, test_images1 / 255.0

#My cnn model, upto the training
#Save upto here.

我可以将执行保存到这里供以后使用吗,其中包括下载的文件训练模型

有!您可以使用 numpy.save("train_images.npy", train_images) 保存 NumPy 数据并使用 train_images = numpy.load("train_images.npy") 加载它们。使用笔记本时,只需将 saveload 放在两个不同的单元格中,然后将 运行 放在您需要的任何单元格中。

文档:

有许多变体,例如 savez 用于将多个数组保存在未压缩的文件中或 savez_compressed 用于压缩文件。

保存模型:

model_json = model.to_json()
with open("model.json", "w") as json_file:
    json_file.write(model_json)
model.save_weights("model.h5")
print("Saved model .......")

加载保存的模型:

json_file = open('model.json', 'r')
loaded_model_json = json_file.read()
json_file.close()
loaded_model = model_from_json(loaded_model_json)

loaded_model.load_weights("model.h5")
print("Loaded model...........")

有关更多详细信息,您可以找到我的实现 here。现在这将同时保存 dataset as well as trained model