Merge pull request #529 from Kimoby/general-improvements-and-bugfixes

General improvements and bugfixes
Neuraxio · Aug 16, 2022 · af917c9 · af917c9
2 parents b95e25f + b70aa53
commit af917c9
Show file tree

Hide file tree

Showing 26 changed files with 828 additions and 494 deletions.
diff --git a/examples/auto_ml/plot_automl_loop_clean_kata.py b/examples/auto_ml/plot_automl_loop_clean_kata.py
@@ -116,7 +116,7 @@ def main(tmpdir: str):
     auto_ml = AutoML(
         pipeline=pipeline,
         hyperparams_optimizer=RandomSearchSampler(),
-        validation_splitter=ValidationSplitter(validation_size=0.20),
+        validation_splitter=ValidationSplitter(validation_size=0.20).set_to_force_expected_outputs_for_scoring(),
         scoring_callback=ScoringCallback(accuracy_score, higher_score_is_better=True),
         n_trials=7,
         epochs=1,