Skip to content

Commit

Permalink
Update Braindecode dependency to 0.8 (fix comment) (#543)
Browse files Browse the repository at this point in the history
  • Loading branch information
PierreGtch authored Feb 2, 2024
1 parent fcfae76 commit 99f6e83
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions examples/plot_braindecode.py
Original file line number Diff line number Diff line change
Expand Up @@ -87,9 +87,10 @@
# In order to create a pipeline, we need to load a model from braindecode.
# the second step is to define a skorch model using EEGClassifier from braindecode
# that allows converting the PyTorch model in a scikit-learn classifier.
# Initialize the model's parameter as a dummy value since this parameter will be set dynamically using the
# callbacks InputShapeSetterEEG, where we have to specify the correct name of the parameter.
# Here, we will use the EEGNet v4 model [1]_ .
# This model has mandatory hyperparameters (the number of channels, the number of classes,
# and the temporal length of the input) but we do not need to specify them because they will
# be set dynamically by EEGClassifier using the input data during the call to the ``.fit()`` method.

# Define a Skorch classifier
clf = EEGClassifier(
Expand Down

0 comments on commit 99f6e83

Please sign in to comment.