[FIX] Refit after validation boolean (#991)

* fix_reverse_refit_with_val * fix_reverse_refit_with_val
Nixtla · May 6, 2024 · f75731b · f75731b
1 parent 2bde2dd
commit f75731b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/nbs/common.base_auto.ipynb b/nbs/common.base_auto.ipynb
@@ -445,7 +445,7 @@
     "            cls_model=self.cls_model,\n",
     "            config=best_config,\n",
     "            dataset=dataset,\n",
-    "            val_size=val_size * (1 - self.refit_with_val),\n",
+    "            val_size=val_size * self.refit_with_val,\n",
     "            test_size=test_size,\n",
     "            distributed_config=distributed_config,\n",
     "        )\n",

diff --git a/neuralforecast/common/_base_auto.py b/neuralforecast/common/_base_auto.py
@@ -425,7 +425,7 @@ def fit(
             cls_model=self.cls_model,
             config=best_config,
             dataset=dataset,
-            val_size=val_size * (1 - self.refit_with_val),
+            val_size=val_size * self.refit_with_val,
             test_size=test_size,
             distributed_config=distributed_config,
         )