update comparison tutorial

Nixtla · Nov 27, 2024 · 55080fb · 55080fb
1 parent 350bd1f
commit 55080fb
Show file tree

Hide file tree

Showing 4 changed files with 712 additions and 4,698 deletions.
diff --git a/nbs/common.base_auto.ipynb b/nbs/common.base_auto.ipynb
@@ -363,7 +363,7 @@
     "                test_size=test_size,\n",
     "                distributed_config=distributed_config,\n",
     "            )\n",
-    "            trial.set_user_attr('config', user_cfg)\n",
+    "            trial.set_user_attr('ALL_PARAMS', user_cfg)\n",
     "            metrics = model.metrics\n",
     "            trial.set_user_attr('METRICS', {\n",
     "                \"loss\": metrics[\"ptl/val_loss\"],\n",
@@ -445,7 +445,7 @@
     "                config=self.config,\n",
     "                distributed_config=distributed_config,\n",
     "            )\n",
-    "            best_config = results.best_trial.user_attrs['config']\n",
+    "            best_config = results.best_trial.user_attrs['ALL_PARAMS']\n",
     "        self.model = self._fit_model(\n",
     "            cls_model=self.cls_model,\n",
     "            config=best_config,\n",

diff --git a/nbs/docs/capabilities/04_hyperparameter_tuning.ipynb b/nbs/docs/capabilities/04_hyperparameter_tuning.ipynb
@@ -1187,7 +1187,7 @@
    ],
    "source": [
     "results = nf.models[0].results.trials_dataframe()\n",
-    "results.drop(columns='user_attrs_config')"
+    "results.drop(columns='user_attrs_ALL_PARAMS')"
    ]
   },
   {