braindecode 0.2.1__tar.gz → 0.3.1__tar.gz

{Braindecode-0.2.1 → Braindecode-0.3.1}/Braindecode.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 1.1
 Name: Braindecode
-Version: 0.2.1
+Version: 0.3.1
 Summary: A deep learning toolbox to decode raw time-domain EEG.
 Home-page: https://github.com/robintibor/braindecode
 Author: Robin Tibor Schirrmeister
@@ -30,16 +30,7 @@ Description: Note: The old braindecode repository has been moved to
         
           pip install numpy
         
-        3. Install newest version of python-mne:
-        
-        .. code-block:: bash
-        
-          git clone git://github.com/mne-tools/mne-python.git
-          cd mne-python
-          python setup.py install
-        
-        
-        4. Install braindecode via pip:
+        3. Install braindecode via pip:
         
         .. code-block:: bash
         

{Braindecode-0.2.1 → Braindecode-0.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 1.1
 Name: Braindecode
-Version: 0.2.1
+Version: 0.3.1
 Summary: A deep learning toolbox to decode raw time-domain EEG.
 Home-page: https://github.com/robintibor/braindecode
 Author: Robin Tibor Schirrmeister
@@ -30,16 +30,7 @@ Description: Note: The old braindecode repository has been moved to
         
           pip install numpy
         
-        3. Install newest version of python-mne:
-        
-        .. code-block:: bash
-        
-          git clone git://github.com/mne-tools/mne-python.git
-          cd mne-python
-          python setup.py install
-        
-        
-        4. Install braindecode via pip:
+        3. Install braindecode via pip:
         
         .. code-block:: bash
         

{Braindecode-0.2.1 → Braindecode-0.3.1}/README.rst

@@ -22,16 +22,7 @@ Installation
 
   pip install numpy
 
-3. Install newest version of python-mne:
-
-.. code-block:: bash
-
-  git clone git://github.com/mne-tools/mne-python.git
-  cd mne-python
-  python setup.py install
-
-
-4. Install braindecode via pip:
+3. Install braindecode via pip:
 
 .. code-block:: bash
 

{Braindecode-0.2.1 → Braindecode-0.3.1}/braindecode/datasets/bcic_iv_2a.py

@@ -33,7 +33,7 @@ class BCICompetition4Set2A(object):
             mask = np.isnan(data[i_chan])
             chan_mean = np.nanmean(data[i_chan])
             data[i_chan, mask] = chan_mean
-        gdf_events = raw_edf.get_edf_events()
+        gdf_events = raw_edf.find_edf_events()
         raw_edf = mne.io.RawArray(data, raw_edf.info, verbose='WARNING')
         # remember gdf events
         raw_edf.info['gdf_events'] = gdf_events
@@ -76,4 +76,4 @@ class BCICompetition4Set2A(object):
         events[:,0] = trial_events[:,0]
         events[:,2] = trial_events[:,1]
 
-        return events, artifact_trial_mask
+        return events, artifact_trial_mask

{Braindecode-0.2.1 → Braindecode-0.3.1}/braindecode/datautil/iterators.py

@@ -162,19 +162,23 @@ class CropsFromTrialsIterator(object):
         Number of predictions ConvNet makes per one input. Can be computed
         by making a forward pass with the given input time length, the
         output length in 3rd dimension is n_preds_per_input.
+    seed: int
+        Random seed for initialization of `numpy.RandomState`.
     
     See Also
     --------
     braindecode.experiments.monitors.compute_preds_per_trial_for_set : Assigns predictions to trials, removes overlaps.
     """
-    def __init__(self, batch_size, input_time_length, n_preds_per_input):
+    def __init__(self, batch_size, input_time_length, n_preds_per_input,
+                 seed=(2017, 6, 28)):
         self.batch_size = batch_size
         self.input_time_length = input_time_length
         self.n_preds_per_input = n_preds_per_input
-        self.rng = RandomState((2017, 6, 28))
+        self.seed = seed
+        self.rng = RandomState(self.seed)
 
     def reset_rng(self):
-        self.rng = RandomState((2017, 6, 28))
+        self.rng = RandomState(self.seed)
 
     def get_batches(self, dataset, shuffle):
         # start always at first predictable sample, so

{Braindecode-0.2.1 → Braindecode-0.3.1}/braindecode/datautil/splitters.py

@@ -77,3 +77,30 @@ def split_into_two_sets(dataset, first_set_fraction=None, n_first_set=None):
     second_set = apply_to_X_y(lambda a: a[n_first_set:], dataset)
     return first_set, second_set
 
+
+def select_examples(dataset, indices):
+    """
+    Select examples from dataset.
+    
+    Parameters
+    ----------
+    dataset: :class:`.SignalAndTarget`
+    indices: list of int, 1d-array of int
+        Indices to select
+
+    Returns
+    -------
+    reduced_set: :class:`.SignalAndTarget`
+        Dataset with only examples selected.
+    """
+    # probably not necessary
+    indices = np.array(indices)
+    if hasattr(dataset.X, 'ndim'):
+        # numpy array
+        new_X = np.array(dataset.X)[indices]
+    else:
+        # list
+        new_X = [dataset.X[i] for i in indices]
+    new_y = np.asarray(dataset.y)[indices]
+    return SignalAndTarget(new_X, new_y)
+