WIP Add TimeLLM

Nixtla · Mar 8, 2024 · 4cd19a1 · 4cd19a1
1 parent 68f09ab
commit 4cd19a1
Show file tree

Hide file tree

Showing 6 changed files with 1,139 additions and 18 deletions.
diff --git a/nbs/core.ipynb b/nbs/core.ipynb
@@ -80,7 +80,7 @@
     "    MLP, NHITS, NBEATS, NBEATSx, DLinear,\n",
     "    TFT, VanillaTransformer,\n",
     "    Informer, Autoformer, FEDformer,\n",
-    "    StemGNN, PatchTST, TimesNet\n",
+    "    StemGNN, PatchTST, TimesNet, TimeLLM\n",
     ")"
    ]
   },
@@ -216,6 +216,7 @@
     "    'tft': TFT, 'autotft': TFT,\n",
     "    'timesnet': TimesNet, 'autotimesnet': TimesNet,\n",
     "    'vanillatransformer': VanillaTransformer, 'autovanillatransformer': VanillaTransformer,\n",
+    "    'timellm': TimeLLM\n",
     "}"
    ]
   },