PyPI - braindecode - Versions diffs - 0.4.0__tar.gz → 0.4.1__tar.gz - Mend

@@ -9,6 +9,20 @@ from braindecode.models.util import to_dense_prediction_model
 class HybridNet(nn.Module, BaseModel):
+    """
+       Hybrid ConvNet model from [3]_.
+       Very hardcoded at the moment.
+       References
+       ----------
+       .. [3] Schirrmeister, R. T., Springenberg, J. T., Fiederer, L. D. J.,
+          Glasstetter, M., Eggensperger, K., Tangermann, M., ... & Ball, T. (2017).
+          Deep learning with convolutional neural networks for EEG decoding and
+          visualization.
+          arXiv preprint arXiv:1703.05051.
+    """
     def __init__(self, n_chans, n_classes, input_time_length):
         super(HybridNet, self).__init__()
         deep_model = Deep4Net(n_chans, n_classes, n_filters_time=20,
@@ -74,4 +88,4 @@ class HybridNet(nn.Module, BaseModel):
         linear_out = self.final_conv(merged_out)
         softmaxed = nn.LogSoftmax(dim=1)(linear_out)
         squeezed = softmaxed.squeeze(3)
-        return squeezed
+        return squeezed

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

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

@@ -137,6 +137,7 @@ class BaseModel(object):
                              "which is the number of timesteps that will be pushed through"
                              "the network in a single pass.")
         if self.cropped:
+            self.network.eval()
             test_input = np_to_var(train_X[0:1], dtype=np.float32)
             while len(test_input.size()) < 4:
                 test_input = test_input.unsqueeze(-1)

@@ -45,7 +45,7 @@ class EEGNetv4(BaseModel):
                  pool_mode='mean',
                  F1=8,
                  D=2,
-                 F2=16,
+                 F2=16, # usually set to F1*D (?)
                  kernel_length=64,
                  third_kernel_size=(8, 4),
                  drop_prob=0.25

@@ -256,7 +256,9 @@
   {
    "cell_type": "code",
    "execution_count": 7,
-   "metadata": {},
+   "metadata": {
+    "collapsed": true
+   },
    "outputs": [],
    "source": [
     "from braindecode.torch_ext.optimizers import AdamW\n",

	@@ -0,0 +1 @@
1	+ __version__ = "0.4.1"

	@@ -1 +0,0 @@
1	- __version__ = "0.4.0"

braindecode 0.4.0__tar.gz → 0.4.1__tar.gz