Keras Прогноз временных рядов с предварительно обученной моделью - PullRequest
1 голос
/ 06 октября 2019

У меня есть заранее определенные временные ряды в Керасе, и я тоже хочу получить прогноз на определенную дату. Вот так:

from keras import load_model
def predict(date, filename):
    model = load_model(filename)
    # Do domething to get input_data from date
    return model.predict(input_data)

Так как это сделать?

1 Ответ

1 голос
/ 06 октября 2019

Вы можете сохранить модель keras с помощью

# assume model is the trained keras model
# save the underlying tensorflow graph
model_file = model.to_json()
with open("model.json", "w") as source:
    source.write(model_file)
# save model parameter 
model.save_weights("weights.h5")

Затем сохраненную модель можно загрузить с помощью

from keras.models import model_from_json

with open("model.json", "r") as f:
    model = model_from_json(f.read())
model.load_weights("weights.h5")
...