Fix bug in MLPClassifer dataset split.

tymorrow · tymorrow · commit c2756bfd288b · 2024-10-30T17:36:23.000-06:00
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)

Original file line number	Diff line number	Diff line change
`@@ -110,7 +110,7 @@ def fit(self, ss: SampleSet, batch_size: int = 200, epochs: int = 20,`
`110`	`110`	`training_dataset = tf.data.Dataset.from_tensor_slices((spectra_tensor, labels_tensor))`
`111`	`111`	`training_dataset, validation_dataset = split_dataset(`
`112`	`112`	`training_dataset,`
`113`		`- left_size=validation_split,`
	`113`	`+ right_size=validation_split,`
`114`	`114`	`shuffle=True`
`115`	`115`	`)`
`116`	`116`	`training_dataset = training_dataset.batch(batch_size=batch_size)`