python - Keras Fit 功能的进度条

Question

我在 Python 3.8.2 中使用 Keras库并使用神经网络 lstm。结束该功能需要很长时间（5天，10天或...）model.fit()。我想在我的应用程序中显示一个进度条，用户会发现他们的需求何时结束。此外，我需要一个将执行顺序保存modlel.fit()在数据库中的函数，让我可以随时向用户显示进度条或终止该modlel.fit()函数。

...
model = Sequential()
model.add(LSTM(4, batch_input_shape=(1, X_train.shape[1], X_train.shape[2]), stateful=True))
model.add(Dense(1))
model.compile(loss='mean_squared_error', optimizer='adam')
model.fit(X_train, y_train,epochs=100, batch_size=1, verbose=1, shuffle=False,callbacks=[call])
model.summary()
model.save("model.h5")
...

标签： pythonmachine-learningkerasneural-networklstm