SliceAndShuffle fix (#43)

AlexanderVNikitin · Mar 26, 2024 · 903bff4 · 903bff4
1 parent b90e021
commit 903bff4
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tsgm/models/augmentations.py b/tsgm/models/augmentations.py
@@ -169,6 +169,8 @@ def generate(self, X: TensorLike, y: Optional[TensorLike] = None, n_samples: int
                     slices.append(s)
                 slices.append(sequence[start_idx:])
                 np.random.shuffle(slices)
+            # concatenate the slices
+            sequence = np.concatenate(slices)
             synthetic_data.append(sequence)
             if has_labels:
                 new_labels.append(y[i])