Added additional NN architectures

NISL-MSU · Jul 25, 2024 · 6f726b0 · 6f726b0
1 parent c849631
commit 6f726b0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/PredictionIntervals/models/NNModel.py b/src/PredictionIntervals/models/NNModel.py
@@ -272,8 +272,8 @@ def trainFold(self, Xtrain, Ytrain, Xval, Yval, batch_size, epochs, filepath, pr
             #     self.basemodel.loadModel(filepath_base)
             # else:
             print("Training base model...")
-            self.basemodel.trainFold(Xtrain.copy(), Ytrain.copy(), Xval.copy(), Yval.copy(), batch_size, int(epochs/2),
-                                     filepath_base, False, yscale, alpha_, False)
+            self.basemodel.trainFold(Xtrain.copy(), Ytrain.copy(), Xval.copy(), Yval.copy(), int(batch_size*2),
+                                     int(epochs/2), filepath_base, False, yscale, alpha_, False)
             print("Base model training complete!")
             time.sleep(0.1)