improve ds tutorial

AlexanderVNikitin · Jun 15, 2024 · 76c94a7 · 76c94a7
1 parent b91cf21
commit 76c94a7
Show file tree

Hide file tree

Showing 3 changed files with 301 additions and 55 deletions.
diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -222,7 +222,6 @@ def test_mmd_3_test():
     "Coffee",
     "ECG200",
     "ElectricDevices",
-    "GunPoint",
     "MixedShapesRegularTrain",
     "StarLightCurves",
     "Wafer"

diff --git a/tsgm/utils/datasets.py b/tsgm/utils/datasets.py
@@ -431,7 +431,7 @@ def get_physionet2012() -> T.Tuple[TensorLike, TensorLike, TensorLike, TensorLik
     This function downloads and retrieves the Physionet 2012 dataset, which consists of physiological data
     and corresponding outcomes. It returns the training, testing, and validation datasets along with their labels.
 
-    :return: A tuple containing the training, testing, and validation datasets along with their labels.
+    :return: A tuple containing the training, testing, and validation datasets along with their labels. (train_X, train_y, test_X, test_y, val_X, val_y)
     :rtype: tuple[TensorLike, TensorLike, TensorLike, TensorLike, TensorLike, TensorLike]
     """
     download_physionet2012()

diff --git a/tutorials/Datasets.ipynb b/tutorials/Datasets.ipynb