keras r如何保存模型并继续训练

Question

我在这里遵循以下示例：https://keras.rstudio.com/articles/examples/lstm_text_generation.html

我正在努力弄清楚如何保存模型，然后在以后继续培训（可能在另一台计算机上）。

谢谢！

Answer 1

keras中的

保存您的模型架构和权重。 然后每次都使用新的输入数据集来加载和拟合模型。

喜欢这种方式。

from keras.layers import SimpleRNN, TimeDistributed

model=Sequential()
model.add(SimpleRNN(input_shape=(None, 2),
                    return_sequences=True, 
                    units=5))
model.add(TimeDistributed(Dense(activation='sigmoid', units=3)))
model.compile(loss = 'mse', optimizer = 'rmsprop')
model.fit(inputs, outputs, epochs = 500, batch_size = 32)

model.save('my_model.h5')

from keras.models import load_model
model = load_model('my_model.h5')

# continue fitting
model.fit(inputs, outputs, epochs = 500, batch_size = 32)