我有一个神经网络,已成功保存在JSON
和h5
文件中。我的问题是,当我从Google Colab保存模型然后尝试将生成的JSON
和h5
文件加载到Jupyter笔记本中时,它不起作用。但是,它确实可以在同一平台上创建和加载它们。我在jupyter和Colab中都运行完全相同的代码。
我这样做是为了保存模型:
model_json = classifier.to_json()
with open("model_test.json", "w") as json_file:
json_file.write(model_json)
# serialize weights to HDF5
classifier.save_weights("model_test.h5")
print("Saved model to disk")
打开它:
json_file = open('model_test.json', 'r')
loaded_model_json = json_file.read()
json_file.close()
loaded_model = model_from_json(loaded_model_json)
# load weights into new model
loaded_model.load_weights("model_test.h5")
print("Loaded model from disk")
如果我留在Jupyter笔记本电脑或Colab中,这些步骤都可以工作,但是我希望能够在Colab中创建一个模型(速度更快),然后在Jupyter笔记本电脑中运行它。