使用 Python 在 lstm 中提前停止

Question

如何在lstm中进行早期停止。

我使用的是 python 张量流，但不是 keras。

如果您能提供示例 python 代码，我将不胜感激。

问候

Answer 1

你可以使用checkpoints：

from keras.callbacks import EarlyStopping
earlyStop=EarlyStopping(monitor="val_loss",verbose=2,mode='min',patience=3)
history=model.fit(xTrain,yTrain,epochs=100,batch_size=10,validation_data=(xTest,yTest) ,verbose=2,callbacks=[earlyStop])

mode='min'即使在 3 个 epoch 后“val_loss”仍未减少（ ），训练也会停止（ patience=3）

#Didn't realize u were note using keras