refactor: more robust model

marcpinet · Apr 24, 2024 · 95eac25 · 95eac25
1 parent 535499b
commit 95eac25
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/neuralnetlib/model.py b/neuralnetlib/model.py
@@ -104,7 +104,7 @@ def fit(self, x_train: np.ndarray, y_train: np.ndarray, epochs: int, batch_size:
             callbacks: List of callback objects (e.g., EarlyStopping)
         """
         x_train = np.array(x_train)
-        x_test = np.array(x_test)
+        y_train = np.array(y_train)
 
         if validation_data is not None:
             x_test, y_test = validation_data