support forecast_fitted_values in distributed

Nixtla · Dec 19, 2023 · a4f88f3 · a4f88f3
1 parent 89f2e02
commit a4f88f3
Show file tree

Hide file tree

Showing 6 changed files with 675 additions and 363 deletions.
diff --git a/nbs/src/core/core.ipynb b/nbs/src/core/core.ipynb
@@ -1908,8 +1908,8 @@
     "            )\n",
     "        assert df is not None\n",
     "        engine = make_execution_engine(infer_by=[df])\n",
-    "        backend = make_backend(engine)\n",
-    "        return backend.forecast(\n",
+    "        self._backend = make_backend(engine)\n",
+    "        return self._backend.forecast(\n",
     "            models=self.models,\n",
     "            fallback_model=self.fallback_model,            \n",
     "            freq=self.freq,            \n",
@@ -1923,6 +1923,13 @@
     "            time_col=time_col,\n",
     "            target_col=target_col,\n",
     "        )\n",
+    "\n",
+    "    def forecast_fitted_values(self):\n",
+    "        if hasattr(self, '_backend'):\n",
+    "            res = self._backend.forecast_fitted_values()\n",
+    "        else:\n",
+    "            res = super().forecast_vitted_values()\n",
+    "        return res\n",
     "    \n",
     "    def cross_validation(\n",
     "        self,\n",