add evaluation tutorial

AlexanderVNikitin · Dec 14, 2023 · 728d81b · 728d81b
1 parent 0a6ca8e
commit 728d81b
Show file tree

Hide file tree

Showing 5 changed files with 848 additions and 2 deletions.
diff --git a/tsgm/dataset.py b/tsgm/dataset.py
@@ -86,8 +86,10 @@ def Xy_concat(self) -> tsgm.types.Tensor:
         elif len(self._y.shape) == 2:
             if self._y.shape[1] == 1:
                 return np.concatenate((self._x, np.repeat(self._y[:, :, None], self._x.shape[1], axis=1)), axis=2)
-            else:
+            elif self._y.shape[1] == self._x.shape[1]:
                 return np.concatenate((self._x, self._y[:, :, None]), axis=2)
+            else:
+                return np.concatenate((self._x, np.repeat(self._y[:, None, :], self._x.shape[1], axis=1)), axis=2)
         else:
             raise ValueError("X & y are not compatible for Xy_concat operation")
 

diff --git a/tutorials/augmentations.ipynb b/tutorials/augmentations.ipynb
@@ -619,7 +619,7 @@
     }
    ],
    "source": [
-    "n, n_ts, n_features  = 1000, 24, 5\n",
+    "n, n_ts, n_features = 1000, 24, 5\n",
     "data = tsgm.utils.gen_sine_dataset(n, n_ts, n_features)\n",
     "scaler = tsgm.utils.TSFeatureWiseScaler()        \n",
     "scaled_data = scaler.fit_transform(data)\n",