eegdash 0.3.1__tar.gz → 0.3.1.dev50__tar.gz

{eegdash-0.3.1 → eegdash-0.3.1.dev50}/LICENSE

@@ -4,6 +4,8 @@ Copyright (C) 2024-2025
 
 Young Truong, UCSD, dt.young112@gmail.com
 Arnaud Delorme, UCSD, adelorme@ucsd.edu
+Aviv Dotan, BGU, avivdot@bgu.post.ac.il
+Oren Shriki, BGU, shrikio@bgu.ac.il
 Bruno Aristimunha, b.aristimunha@gmail.com
 
 This program is free software; you can redistribute it and/or modify

{eegdash-0.3.1 → eegdash-0.3.1.dev50}/PKG-INFO

@@ -1,14 +1,16 @@
 Metadata-Version: 2.4
 Name: eegdash
-Version: 0.3.1
+Version: 0.3.1.dev50
 Summary: EEG data for machine learning
-Author-email: Young Truong <dt.young112@gmail.com>, Arnaud Delorme <adelorme@gmail.com>, Bruno Aristimunha <b.aristimunha@gmail.com>
+Author-email: Young Truong <dt.young112@gmail.com>, Arnaud Delorme <adelorme@gmail.com>, Aviv Dotan <avivd220@gmail.com>, Oren Shriki <oren70@gmail.com>, Bruno Aristimunha <b.aristimunha@gmail.com>
 License: GNU General Public License
         
         Copyright (C) 2024-2025
         
         Young Truong, UCSD, dt.young112@gmail.com
         Arnaud Delorme, UCSD, adelorme@ucsd.edu
+        Aviv Dotan, BGU, avivdot@bgu.post.ac.il
+        Oren Shriki, BGU, shrikio@bgu.ac.il
         Bruno Aristimunha, b.aristimunha@gmail.com
         
         This program is free software; you can redistribute it and/or modify

{eegdash-0.3.1 → eegdash-0.3.1.dev50}/eegdash/__init__.py

@@ -5,4 +5,4 @@ from .utils import __init__mongo_client
 __init__mongo_client()
 
 __all__ = ["EEGDash", "EEGDashDataset", "EEGChallengeDataset"]
-__version__ = "0.3.1"
+__version__ = "0.3.1.dev50"

{eegdash-0.3.1 → eegdash-0.3.1.dev50}/eegdash/data_utils.py

@@ -55,21 +55,25 @@ class EEGDashBaseDataset(BaseDataset):
         self.cache_dir = Path(cache_dir)
         bids_kwargs = self.get_raw_bids_args()
 
-        self.bidspath = BIDSPath(
-            root=self.cache_dir / record["dataset"],
-            datatype="eeg",
-            suffix="eeg",
-            **bids_kwargs,
-        )
         if s3_bucket:
             self.s3_bucket = s3_bucket
             self.s3_open_neuro = False
+            bids_root = self.cache_dir
+            self.filecache = self.cache_dir / record["bidspath"]
         else:
             self.s3_bucket = self._AWS_BUCKET
             self.s3_open_neuro = True
+            bids_root = self.cache_dir / record["dataset"]
+            self.filecache = self.cache_dir / record["bidspath"]
+
+        self.bidspath = BIDSPath(
+            root=bids_root,
+            datatype="eeg",
+            suffix="eeg",
+            **bids_kwargs,
+        )
 
         self.s3file = self.get_s3path(record["bidspath"])
-        self.filecache = self.cache_dir / record["bidspath"]
         self.bids_dependencies = record["bidsdependencies"]
         # Temporary fix for BIDS dependencies path
         # just to release to the competition
@@ -85,14 +89,18 @@ class EEGDashBaseDataset(BaseDataset):
         return f"{self.s3_bucket}/{filepath}"
 
     def _download_s3(self) -> None:
-        """Fetch the given data from its S3 location and cache it locally."""
-        self.filecache.parent.mkdir(parents=True, exist_ok=True)
+        """Download function that gets the raw EEG data from S3."""
         filesystem = s3fs.S3FileSystem(
             anon=True, client_kwargs={"region_name": "us-east-2"}
         )
         if not self.s3_open_neuro:
             self.s3file = re.sub(r"(^|/)ds\d{6}/", r"\1", self.s3file, count=1)
+            self.filecache = re.sub(
+                r"(^|/)ds\d{6}/", r"\1", str(self.filecache), count=1
+            )
+            self.filecache = Path(self.filecache)
 
+        self.filecache.parent.mkdir(parents=True, exist_ok=True)
         filesystem.download(self.s3file, self.filecache)
         self.filenames = [self.filecache]
 
@@ -106,6 +114,8 @@ class EEGDashBaseDataset(BaseDataset):
         for dep in self.bids_dependencies:
             s3path = self.get_s3path(dep)
             filepath = self.cache_dir / dep
+            # here, we download the dependency and it is fine
+            # in the case of the competition.
             if not filepath.exists():
                 filepath.parent.mkdir(parents=True, exist_ok=True)
                 filesystem.download(s3path, filepath)

{eegdash-0.3.1 → eegdash-0.3.1.dev50}/eegdash/features/feature_bank/complexity.py

@@ -71,7 +71,7 @@ def complexity_svd_entropy(x, m=10, tau=1):
 @FeaturePredecessor(*SIGNAL_PREDECESSORS)
 @univariate_feature
 @nb.njit(cache=True, fastmath=True)
-def complexity_lempel_ziv(x, threshold=None):
+def complexity_lempel_ziv(x, threshold=None, normalize=True):
     lzc = np.empty(x.shape[:-1])
     for i in np.ndindex(x.shape[:-1]):
         t = np.median(x[i]) if threshold is None else threshold
@@ -97,4 +97,6 @@ def complexity_lempel_ziv(x, threshold=None):
                     j, k, k_max = 0, 1, 1
                 else:
                     k = 1
+        if normalize:
+            lzc[i] *= np.log2(n) / n
     return lzc

{eegdash-0.3.1 → eegdash-0.3.1.dev50}/eegdash/features/feature_bank/csp.py

@@ -97,5 +97,5 @@ class CommonSpatialPattern(TrainableFeature):
                 + "all weights were filtered out."
             )
         proj = (self.transform_input(x) - self._mean) @ w
-        proj = proj.reshape(x.shape[0], x.shape[2], -1).mean(axis=1)
+        proj = proj.reshape(x.shape[0], x.shape[2], -1).var(axis=1)
         return {f"{i}": proj[:, i] for i in range(proj.shape[-1])}

{eegdash-0.3.1 → eegdash-0.3.1.dev50}/eegdash.egg-info/PKG-INFO

@@ -1,14 +1,16 @@
 Metadata-Version: 2.4
 Name: eegdash
-Version: 0.3.1
+Version: 0.3.1.dev50
 Summary: EEG data for machine learning
-Author-email: Young Truong <dt.young112@gmail.com>, Arnaud Delorme <adelorme@gmail.com>, Bruno Aristimunha <b.aristimunha@gmail.com>
+Author-email: Young Truong <dt.young112@gmail.com>, Arnaud Delorme <adelorme@gmail.com>, Aviv Dotan <avivd220@gmail.com>, Oren Shriki <oren70@gmail.com>, Bruno Aristimunha <b.aristimunha@gmail.com>
 License: GNU General Public License
         
         Copyright (C) 2024-2025
         
         Young Truong, UCSD, dt.young112@gmail.com
         Arnaud Delorme, UCSD, adelorme@ucsd.edu
+        Aviv Dotan, BGU, avivdot@bgu.post.ac.il
+        Oren Shriki, BGU, shrikio@bgu.ac.il
         Bruno Aristimunha, b.aristimunha@gmail.com
         
         This program is free software; you can redistribute it and/or modify

{eegdash-0.3.1 → eegdash-0.3.1.dev50}/pyproject.toml

@@ -8,6 +8,8 @@ name = "eegdash"
 authors = [
   { name = "Young Truong",    email = "dt.young112@gmail.com" },
   { name = "Arnaud Delorme",  email = "adelorme@gmail.com" },
+  { name = "Aviv Dotan",  email = "avivd220@gmail.com" },
+  { name = "Oren Shriki",  email = "oren70@gmail.com" },
   { name = "Bruno Aristimunha", email = "b.aristimunha@gmail.com"}
 ]
 description     = "EEG data for machine learning"