У меня есть нейронная сеть, которую я успешно сохранил в 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.