From c2756bfd288ba5cf038399bad239563dd4e34cc8 Mon Sep 17 00:00:00 2001 From: Tyler Morrow Date: Wed, 30 Oct 2024 17:25:44 -0600 Subject: [PATCH] Fix bug in MLPClassifer dataset split. --- riid/models/neural_nets/basic.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/riid/models/neural_nets/basic.py b/riid/models/neural_nets/basic.py index 705b32e..e416947 100644 --- 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)