Fix bug in MLPClassifer dataset split.

sandialabs · Oct 30, 2024 · c2756bf · c2756bf
1 parent b4ddd1a
commit c2756bf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/riid/models/neural_nets/basic.py b/riid/models/neural_nets/basic.py
@@ -110,7 +110,7 @@ def fit(self, ss: SampleSet, batch_size: int = 200, epochs: int = 20,
         training_dataset = tf.data.Dataset.from_tensor_slices((spectra_tensor, labels_tensor))
         training_dataset, validation_dataset = split_dataset(
             training_dataset,
-            left_size=validation_split,
+            right_size=validation_split,
             shuffle=True
         )
         training_dataset = training_dataset.batch(batch_size=batch_size)