diff --git a/nbs/core.ipynb b/nbs/core.ipynb index ba1141ca6..c59bbe7a3 100644 --- a/nbs/core.ipynb +++ b/nbs/core.ipynb @@ -93,7 +93,7 @@ " StemGNN, PatchTST, TimesNet, TimeLLM, TSMixer, TSMixerx,\n", " MLPMultivariate, iTransformer,\n", " BiTCN, TiDE, DeepNPTS, SOFTS,\n", - " TimeMixer, KAN\n", + " TimeMixer, KAN, RMoK\n", ")\n", "from neuralforecast.common._base_auto import BaseAuto, MockTrial" ] @@ -245,7 +245,8 @@ " 'deepnpts': DeepNPTS, 'autodeepnpts': DeepNPTS,\n", " 'softs': SOFTS, 'autosofts': SOFTS,\n", " 'timemixer': TimeMixer, 'autotimemixer': TimeMixer,\n", - " 'kan': KAN, 'autokan': KAN\n", + " 'kan': KAN, 'autokan': KAN,\n", + " 'rmok': RMoK, 'autormok': RMoK\n", "}" ] }, diff --git a/neuralforecast/core.py b/neuralforecast/core.py index a18127f51..b21815b69 100644 --- a/neuralforecast/core.py +++ b/neuralforecast/core.py @@ -66,6 +66,7 @@ SOFTS, TimeMixer, KAN, + RMoK, ) from .common._base_auto import BaseAuto, MockTrial @@ -190,6 +191,8 @@ def _insample_times( "autotimemixer": TimeMixer, "kan": KAN, "autokan": KAN, + "rmok": RMoK, + "autormok": RMoK, } # %% ../nbs/core.ipynb 8