fix_comments_CC

Nixtla · Mar 22, 2024 · e5d7b9e · e5d7b9e
1 parent a5e6a9b
commit e5d7b9e
Show file tree

Hide file tree

Showing 3 changed files with 290 additions and 11 deletions.
diff --git a/nbs/examples/Multivariate_with_TSMixer.ipynb b/nbs/examples/Multivariate_with_TSMixer.ipynb
diff --git a/nbs/models.tsmixerx.ipynb b/nbs/models.tsmixerx.ipynb
@@ -466,7 +466,6 @@
     "                          self.loss.outputsize_multiplier,\n",
     "                          -1)                                   #   [B, h, N * n_outputs] -> [B, h, n_outputs, N]\n",
     "            x = self.norm.reverse(x)\n",
-    "            # x = x.reshape(batch_size, self.h, -1)               #   [B, h, n_outputs, N] -> [B, h, n_outputs * N]\n",
     "            x = x.reshape(batch_size, self.h, -1)               #   [B, h, n_outputs, N] -> [B, h, n_outputs * N]\n",
     "\n",
     "        # Map to loss domain\n",

diff --git a/neuralforecast/models/tsmixerx.py b/neuralforecast/models/tsmixerx.py
@@ -415,7 +415,6 @@ def forward(self, windows_batch):
                 batch_size, self.h, self.loss.outputsize_multiplier, -1
             )  #   [B, h, N * n_outputs] -> [B, h, n_outputs, N]
             x = self.norm.reverse(x)
-            # x = x.reshape(batch_size, self.h, -1)               #   [B, h, n_outputs, N] -> [B, h, n_outputs * N]
             x = x.reshape(
                 batch_size, self.h, -1
             )  #   [B, h, n_outputs, N] -> [B, h, n_outputs * N]