Merge pull request #36 from IBM/ttmfix

ttm fix
ibm-granite · Apr 9, 2024 · 8c8e54e · 8c8e54e
2 parents cfa2e1e + 4ec5473
commit 8c8e54e
Show file tree

Hide file tree

Showing 10 changed files with 242 additions and 96 deletions.
diff --git a/notebooks/hfdemo/tinytimemixer/ttm_benchmarking_1024_96.ipynb b/notebooks/hfdemo/tinytimemixer/ttm_benchmarking_1024_96.ipynb
@@ -1,6 +1,7 @@
 {
  "cells": [
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -10,6 +11,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -43,7 +45,7 @@
     "\n",
     "# First Party\n",
     "from tsfm_public.toolkit.callbacks import TrackingCallback\n",
-    "from notebooks.hfdemo.tinytimemixer.utils import (\n",
+    "from tsfm_public.models.tinytimemixer.utils import (\n",
     "    count_parameters,\n",
     "    get_data,\n",
     "    plot_preds,\n",
@@ -54,6 +56,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -91,6 +94,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -115,6 +119,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -139,6 +144,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -354,6 +360,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -541,6 +548,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -550,27 +558,31 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
     "### Comparing forecasting error (MSE)"
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
     "![TTM_vs_MOIRAI](images/ttm_moirai.png)"
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
     "### Comparing model size (in Millions)"
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -583,13 +595,15 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
     "## Conclusion"
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [

diff --git a/notebooks/hfdemo/tinytimemixer/ttm_benchmarking_512_96.ipynb b/notebooks/hfdemo/tinytimemixer/ttm_benchmarking_512_96.ipynb
@@ -1,6 +1,7 @@
 {
  "cells": [
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -10,6 +11,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -43,7 +45,7 @@
     "\n",
     "# First Party\n",
     "from tsfm_public.toolkit.callbacks import TrackingCallback\n",
-    "from notebooks.hfdemo.tinytimemixer.utils import (\n",
+    "from tsfm_public.models.tinytimemixer.utils import (\n",
     "    count_parameters,\n",
     "    get_data,\n",
     "    plot_preds,\n",
@@ -54,6 +56,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -91,6 +94,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -115,6 +119,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -139,6 +144,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -354,6 +360,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [

diff --git a/notebooks/hfdemo/tinytimemixer/ttm_m4_hourly.ipynb b/notebooks/hfdemo/tinytimemixer/ttm_m4_hourly.ipynb
@@ -1,6 +1,7 @@
 {
  "cells": [
   {
+   "attachments": {},
    "cell_type": "markdown",
    "id": "259c89f5-45b5-4fe4-9034-21abd3e6328a",
    "metadata": {},
@@ -62,7 +63,7 @@
     "\n",
     "# First Party\n",
     "from tsfm_public.toolkit.callbacks import TrackingCallback\n",
-    "from notebooks.hfdemo.tinytimemixer.utils import (\n",
+    "from tsfm_public.models.tinytimemixer.utils import (\n",
     "    count_parameters,\n",
     "    get_data,\n",
     "    plot_preds,\n",
@@ -100,6 +101,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "id": "bea706a0-6913-4395-b7fc-e05189cff3e4",
    "metadata": {},
@@ -111,6 +113,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "id": "0a099f2d-32c8-4420-ad05-57ad7f6ad3e7",
    "metadata": {},
@@ -172,6 +175,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "id": "2061f3a7-05c0-413a-af91-96a156f4c962",
    "metadata": {},
@@ -330,6 +334,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "id": "025ef9f5-93de-4c4c-aa8a-d5e377936be3",
    "metadata": {},
@@ -390,6 +395,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "id": "817c2f37-2233-46f5-9981-d2b1de4ab63d",
    "metadata": {},
@@ -436,6 +442,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "id": "0bd907db-c6db-488f-bb68-169bbdd054ed",
    "metadata": {},
@@ -735,6 +742,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "id": "bfbe886a-96d9-4abf-ae52-f3699ce5c6bc",
    "metadata": {},
@@ -950,6 +958,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "id": "c256b55f-dcff-4fef-9e1c-28a1fa58e281",
    "metadata": {},
@@ -995,6 +1004,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "id": "a7fcc53b-71f5-480a-a0e3-5e614203dbfb",
    "metadata": {},
@@ -1088,6 +1098,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "id": "974ae281-08a1-4ed6-95a9-01ff47dd88cf",
    "metadata": {},
@@ -1096,6 +1107,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "id": "d4103272-e62d-4310-bb93-4613fe21d56f",
    "metadata": {},
@@ -1124,6 +1136,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "id": "0eea17c2-c9d1-4f70-8126-7b078afff23c",
    "metadata": {},
@@ -1250,6 +1263,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "id": "a3e3f668-2998-425d-b8e2-732da6903366",
    "metadata": {},
@@ -1268,6 +1282,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "id": "98f78413-3c45-4aa4-b297-f9619254866a",
    "metadata": {},
@@ -1346,6 +1361,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "id": "818446de-b2aa-4477-bdd0-982affeac6de",
    "metadata": {},
@@ -1356,6 +1372,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "id": "dcfba24c-57ca-4e59-98b2-88cad3261df2",
    "metadata": {},
@@ -1364,6 +1381,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "id": "71d3a388-71a3-498e-8912-8a28ba02dd03",
    "metadata": {},
@@ -1377,6 +1395,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "id": "4703aba5-9e6d-4045-bcfe-65fffa962d03",
    "metadata": {},
@@ -1385,6 +1404,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "id": "498765c6-620f-4153-b6d4-c767af6aee04",
    "metadata": {},
@@ -1398,6 +1418,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "id": "b9bc4134-f1ad-40a2-aa7f-cd65cf7eb8c7",
    "metadata": {},
@@ -1408,6 +1429,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "id": "d5b87661-f042-4705-a6e7-b24732700e61",
    "metadata": {},