braindecode 0.1.4.2__tar.gz → 0.1.4.4__tar.gz

{Braindecode-0.1.4.2 → Braindecode-0.1.4.4}/Braindecode.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 1.1
 Name: Braindecode
-Version: 0.1.4.2
+Version: 0.1.4.4
 Summary: A deep learning toolbox to decode raw time-domain EEG.
 Home-page: https://github.com/robintibor/braindecode
 Author: Robin Tibor Schirrmeister

{Braindecode-0.1.4.2 → Braindecode-0.1.4.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 1.1
 Name: Braindecode
-Version: 0.1.4.2
+Version: 0.1.4.4
 Summary: A deep learning toolbox to decode raw time-domain EEG.
 Home-page: https://github.com/robintibor/braindecode
 Author: Robin Tibor Schirrmeister

{Braindecode-0.1.4.2 → Braindecode-0.1.4.4}/braindecode/datautil/iterators.py

@@ -77,7 +77,13 @@ class BalancedBatchSizeIterator(object):
                                        rng=self.rng,
                                        shuffle=shuffle)
         for batch_inds in batches:
-            yield (dataset.X[batch_inds], dataset.y[batch_inds])
+            batch_X = dataset.X[batch_inds]
+            batch_y = dataset.y[batch_inds]
+
+            # add empty fourth dimension if necessary
+            if batch_X.ndim == 3:
+                batch_X = batch_X[:, :, :, None]
+            yield (batch_X, batch_y)
 
     def reset_rng(self):
         self.rng = RandomState(328774)

{Braindecode-0.1.4.2 → Braindecode-0.1.4.4}/braindecode/datautil/trial_segment.py

@@ -59,10 +59,10 @@ def create_signal_target_from_start_and_ival(
     y = []
 
     for i_sample, mrk_code in zip(events[:, 0], events[:, 1]):
-        start_sample = i_sample + start_offset
-        stop_sample = i_sample + stop_offset
-        name, this_y = mrk_code_to_name_and_y[mrk_code]
+        start_sample = int(i_sample) + start_offset
+        stop_sample = int(i_sample) + stop_offset
         if mrk_code in mrk_code_to_name_and_y:
+            name, this_y = mrk_code_to_name_and_y[mrk_code]
             X.append(data[:, start_sample:stop_sample].astype(np.float32))
             y.append(np.int64(this_y))
             class_to_n_trials[name] += 1

{Braindecode-0.1.4.2 → Braindecode-0.1.4.4}/docs/notebooks/Experiment_Class.ipynb

@@ -41,7 +41,7 @@
     "\n",
     "<div class='alert alert-warning'>\n",
     "\n",
-    "It is not necessary to use the Experiment class to use other functionality of Braindecode. Feel free to ignore it :)\n",
+    "It is not necessary to use the Experiment class to use the other functionality of Braindecode. Feel free to ignore it :)\n",
     "\n",
     "</div>"
    ]

{Braindecode-0.1.4.2 → Braindecode-0.1.4.4}/setup.py

@@ -15,7 +15,7 @@ setup(
     # Versions should comply with PEP440.  For a discussion on single-sourcing
     # the version across setup.py and the project code, see
     # http://packaging.python.org/en/latest/tutorial.html#version
-    version='0.1.4.02',
+    version='0.1.4.04',
 
     description='A deep learning toolbox to decode raw time-domain EEG.',
     long_description=long_description,