Added a comment to project 1

Author: Magnuti

project_1/neural_network.py

@@ -166,7 +166,7 @@ def batch_loader(self, X: np.ndarray, Y: np.ndarray, batch_size: int, shuffle=Tr
             num_batches = dataset_size // batch_size
         else:
             # Drop last batch if the dataset is not evenly divisible by the batch size
-            num_batches = int(np.ceil(dataset_size / batch_size))
+            num_batches = int(np.ceil(dataset_size / batch_size)) # TODO .floor() maybe?
 
         indices = list(range(dataset_size))