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

ruancomelli · ruancomelli · commit 6c675326b5f0 · 2022-09-17T12:10:50.000-03:00
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)),
     )

Original file line number	Diff line number	Diff line change
`@@ -39,5 +39,5 @@ def fit_hypermodel(`
`39`	`39`
`40`	`40`	`return TuneModelReturn(`
`41`	`41`	`model=model,`
`42`		`- evaluation=model.evaluate(ds_val),`
	`42`	`+ evaluation=model.evaluate(ds_val.batch(params.batch_size)),`
`43`	`43`	`)`