Changed early stopping

NISL-MSU · Jul 19, 2024 · b7b8c20 · b7b8c20
1 parent c73a9c6
commit b7b8c20
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/PredictionIntervals/models/NNModel.py b/src/PredictionIntervals/models/NNModel.py
@@ -459,7 +459,7 @@ def trainFold(self, Xtrain, Ytrain, Xval, Yval, batch_size, epochs, filepath, pr
             elif self.method == 'DualAQD' and epoch > 1000:
                 if not improved:
                     cnt += 1
-                    if cnt == 2000:
+                    if cnt == 2500:
                         print("Early stopping at epoch: ", epoch)
                         break
                 else: