braindecode 0.1.5__tar.gz → 0.1.5.2__tar.gz

{Braindecode-0.1.5 → Braindecode-0.1.5.2}/Braindecode.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 1.1
 Name: Braindecode
-Version: 0.1.5
+Version: 0.1.5.2
 Summary: A deep learning toolbox to decode raw time-domain EEG.
 Home-page: https://github.com/robintibor/braindecode
 Author: Robin Tibor Schirrmeister
@@ -22,7 +22,13 @@ Description: Note: The old braindecode repository has been moved to
         
         1. Install pytorch from http://pytorch.org/ (you don't need to install torchvision).
         
-        2. Install newest version of python-mne:
+        2. Install numpy (necessary for resamply installation to work), e.g.:
+        
+        .. code-block:: bash
+        
+          pip install numpy
+        
+        3. Install newest version of python-mne:
         
         .. code-block:: bash
         
@@ -31,7 +37,7 @@ Description: Note: The old braindecode repository has been moved to
           python setup.py install
         
         
-        3. Install braindecode via pip:
+        4. Install braindecode via pip:
         
         .. code-block:: bash
         

{Braindecode-0.1.5 → Braindecode-0.1.5.2}/Braindecode.egg-info/SOURCES.txt

@@ -41,7 +41,9 @@ braindecode/torch_ext/util.py
 docs/Makefile
 docs/conf.py
 docs/index.rst
+docs/notebooks/BBCI_Comparison.ipynb
 docs/notebooks/BBCI_Data.ipynb
+docs/notebooks/BBCI_Data_Common_Average?.ipynb
 docs/notebooks/BBCI_Data_Start_Stop.ipynb
 docs/notebooks/BBCI_Data_Trialwise.ipynb
 docs/notebooks/Cropped_Decoding.ipynb

{Braindecode-0.1.5 → Braindecode-0.1.5.2}/Braindecode.egg-info/requires.txt

@@ -5,4 +5,3 @@ scipy
 resampy
 matplotlib
 h5py
-torch

{Braindecode-0.1.5 → Braindecode-0.1.5.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 1.1
 Name: Braindecode
-Version: 0.1.5
+Version: 0.1.5.2
 Summary: A deep learning toolbox to decode raw time-domain EEG.
 Home-page: https://github.com/robintibor/braindecode
 Author: Robin Tibor Schirrmeister
@@ -22,7 +22,13 @@ Description: Note: The old braindecode repository has been moved to
         
         1. Install pytorch from http://pytorch.org/ (you don't need to install torchvision).
         
-        2. Install newest version of python-mne:
+        2. Install numpy (necessary for resamply installation to work), e.g.:
+        
+        .. code-block:: bash
+        
+          pip install numpy
+        
+        3. Install newest version of python-mne:
         
         .. code-block:: bash
         
@@ -31,7 +37,7 @@ Description: Note: The old braindecode repository has been moved to
           python setup.py install
         
         
-        3. Install braindecode via pip:
+        4. Install braindecode via pip:
         
         .. code-block:: bash
         

{Braindecode-0.1.5 → Braindecode-0.1.5.2}/README.rst

@@ -14,7 +14,13 @@ Installation
 
 1. Install pytorch from http://pytorch.org/ (you don't need to install torchvision).
 
-2. Install newest version of python-mne:
+2. Install numpy (necessary for resamply installation to work), e.g.:
+
+.. code-block:: bash
+
+  pip install numpy
+
+3. Install newest version of python-mne:
 
 .. code-block:: bash
 
@@ -23,7 +29,7 @@ Installation
   python setup.py install
 
 
-3. Install braindecode via pip:
+4. Install braindecode via pip:
 
 .. code-block:: bash
 

{Braindecode-0.1.5 → Braindecode-0.1.5.2}/braindecode/models/eegnet.py

@@ -46,7 +46,7 @@ class EEGNet(object):
         model = nn.Sequential()
         n_filters_1 = 16
         model.add_module('conv_1', nn.Conv2d(
-            self.in_chans, n_filters_1, (1, 1), stride=1, bias=False))
+            self.in_chans, n_filters_1, (1, 1), stride=1, bias=True))
         model.add_module('bnorm_1', nn.BatchNorm2d(
             n_filters_1, momentum=0.1, affine=True),)
         model.add_module('elu_1', Expression(elu))
@@ -60,7 +60,7 @@ class EEGNet(object):
         model.add_module('conv_2', nn.Conv2d(
             1, n_filters_2, self.second_kernel_size, stride=1,
             padding=(self.second_kernel_size[0] // 2, 0),
-            bias=False))
+            bias=True))
         model.add_module('bnorm_2',nn.BatchNorm2d(
             n_filters_2, momentum=0.1, affine=True),)
         model.add_module('elu_2', Expression(elu))
@@ -72,7 +72,7 @@ class EEGNet(object):
         model.add_module('conv_3', nn.Conv2d(
             n_filters_2, n_filters_3, self.third_kernel_size, stride=1,
             padding=(self.third_kernel_size[0] // 2, 0),
-            bias=False))
+            bias=True))
         model.add_module('bnorm_3',nn.BatchNorm2d(
             n_filters_3, momentum=0.1, affine=True),)
         model.add_module('elu_3', Expression(elu))

{Braindecode-0.1.5 → Braindecode-0.1.5.2}/braindecode/torch_ext/losses.py

@@ -29,3 +29,13 @@ def log_categorical_crossentropy(logpreds, targets, dims=None):
         for dim in dims:
             result = th.sum(result, dim=int(dim))
     return result
+
+
+def l2_loss(model):
+    losses = [th.sum(p * p) for p in model.parameters()]
+    return sum(losses)
+
+
+def l1_loss(model):
+    losses = [th.sum(th.abs(p)) for p in model.parameters()]
+    return sum(losses)

{Braindecode-0.1.5 → Braindecode-0.1.5.2}/docs/index.rst

@@ -11,7 +11,13 @@ Installation
 
 1. Install pytorch from http://pytorch.org/ (you don't need to install torchvision).
 
-2. Install newest version of python-mne:
+2. Install numpy (necessary for resamply installation to work), e.g.:
+
+.. code-block:: bash
+
+  pip install numpy
+
+3. Install newest version of python-mne:
 
 .. code-block:: bash
 
@@ -20,7 +26,7 @@ Installation
   python setup.py install
 
 
-3. Install braindecode via pip:
+4. Install braindecode via pip:
 
 .. code-block:: bash