Use training set instead of full dataset to train

RubixML · Jul 8, 2020 · fc11c81 · fc11c81
1 parent 185cd4b
commit fc11c81
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train.php b/train.php
@@ -17,7 +17,7 @@
 
 echo 'Training ...' . PHP_EOL;
 
-$estimator->train($dataset);
+$estimator->train($training);
 
 echo 'Making predictions ...' . PHP_EOL;