fix(automl/tuning): ensure that the valset is batched when evaluating

ruancomelli · Sep 17, 2022 · 6c67532 · 6c67532
1 parent 0bb4d1f
commit 6c67532
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/boiling_learning/automl/tuning.py b/boiling_learning/automl/tuning.py
@@ -39,5 +39,5 @@ def fit_hypermodel(
 
     return TuneModelReturn(
         model=model,
-        evaluation=model.evaluate(ds_val),
+        evaluation=model.evaluate(ds_val.batch(params.batch_size)),
     )