Merge pull request #65 from ajati/plot_fix

Utility plot fixes
ibm-granite · Jun 11, 2024 · b5bc3ba · b5bc3ba
2 parents f52ebf9 + ef7853c
commit b5bc3ba
Show file tree

Hide file tree

Showing 5 changed files with 997 additions and 496 deletions.
diff --git a/notebooks/hfdemo/tinytimemixer/ttm_m4_hourly.ipynb b/notebooks/hfdemo/tinytimemixer/ttm_m4_hourly.ipynb
@@ -61,14 +61,6 @@
     "import numpy as np\n",
     "import pandas as pd\n",
     "\n",
-    "# First Party\n",
-    "from tsfm_public.toolkit.callbacks import TrackingCallback\n",
-    "from tsfm_public.models.tinytimemixer.utils import (\n",
-    "    count_parameters,\n",
-    "    get_data,\n",
-    "    plot_preds,\n",
-    ")\n",
-    "\n",
     "# Local\n",
     "from tsfm_public.models.tinytimemixer import TinyTimeMixerForPrediction"
    ]

diff --git a/notebooks/hfdemo/ttm_getting_started.ipynb b/notebooks/hfdemo/ttm_getting_started.ipynb