increase windows and samples

Nixtla · Nov 20, 2024 · 358e81b · 358e81b
1 parent 259f2cb
commit 358e81b
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/nbs/auto.ipynb b/nbs/auto.ipynb
@@ -1808,6 +1808,13 @@
    "source": [
     "#| hide\n",
     "# using input_size\n",
+    "fit_kwargs = dict(\n",
+    "    n_windows=3,\n",
+    "    h=h,\n",
+    "    num_samples=5,\n",
+    "    optimize_kwargs={'timeout': 60},    \n",
+    ")\n",
+    "\n",
     "start = time.perf_counter()\n",
     "auto_mlf.fit(df=train, **fit_kwargs)\n",
     "no_limit = time.perf_counter() - start\n",