Note
Go to the end to download the full example code.
Cross-session motor imagery with deep learning EEGNet v4 model#
This example shows how to use BrainDecode in combination with MOABB evaluation. In this example, we use the architecture EEGNet.
# Authors: Igor Carrara <igor.carrara@inria.fr>
# Bruno Aristimunha <b.aristimunha@gmail.com>
#
# License: BSD (3-clause)
import matplotlib.pyplot as plt
import mne
import seaborn as sns
import torch
from moabb.datasets import BNCI2014_001
from moabb.evaluations import CrossSessionEvaluation
from moabb.paradigms import MotorImagery
from moabb.utils import setup_seed
from sklearn.pipeline import make_pipeline
from skorch.callbacks import EarlyStopping, EpochScoring
from skorch.dataset import ValidSplit
from braindecode import EEGClassifier
from braindecode.models import EEGNet
mne.set_log_level(False)
# Print Information PyTorch
print(f"Torch Version: {torch.__version__}")
# Set up GPU if it is there
cuda = torch.cuda.is_available()
device = "cuda" if cuda else "cpu"
print("GPU is", "AVAILABLE" if cuda else "NOT AVAILABLE")
Torch Version: 2.14.1+cpu
GPU is NOT AVAILABLE
In this example, we will use only the dataset BNCI2014_001.
Running the benchmark#
This example uses the CrossSession evaluation procedure. We focus on the dataset BNCI2014_001 and only on 1 subject to reduce computational time.
To keep the computational time low, the epoch is reduced. In a real situation, we suggest using the following: EPOCH = 1000 PATIENCE = 300
This code is implemented to run on the CPU. If you’re using a GPU, do not use multithreading (i.e. set n_jobs=1)
# Set random seed to be able to reproduce results
seed = 42
setup_seed(seed)
# Ensure that all operations are deterministic on GPU (if used) for reproducibility
torch.backends.cudnn.deterministic = True
torch.backends.cudnn.benchmark = False
# Hyperparameter
LEARNING_RATE = 0.0625 * 0.01 # parameter taken from Braindecode
WEIGHT_DECAY = 0 # parameter taken from Braindecode
BATCH_SIZE = 64 # parameter taken from BrainDecode
EPOCH = 10
PATIENCE = 3
fmin = 4
fmax = 100
tmin = 0
tmax = None
# Load the dataset
dataset = BNCI2014_001()
events = ["right_hand", "left_hand"]
paradigm = MotorImagery(
events=events, n_classes=len(events), fmin=fmin, fmax=fmax, tmin=tmin, tmax=tmax
)
subjects = [1]
X, _, _ = paradigm.get_data(dataset=dataset, subjects=subjects)
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
Create Pipelines#
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.
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(
module=EEGNet,
optimizer=torch.optim.Adam,
optimizer__lr=LEARNING_RATE,
batch_size=BATCH_SIZE,
max_epochs=EPOCH,
train_split=ValidSplit(0.2, random_state=seed),
device=device,
callbacks=[
EarlyStopping(monitor="valid_loss", patience=PATIENCE),
EpochScoring(
scoring="accuracy", on_train=True, name="train_acc", lower_is_better=False
),
EpochScoring(
scoring="accuracy", on_train=False, name="valid_acc", lower_is_better=False
),
],
verbose=1, # Not printing the results for each epoch
)
# Create the pipelines
pipes = {}
pipes["EEGNet"] = make_pipeline(clf)
Evaluation#
dataset.subject_list = dataset.subject_list[:2]
evaluation = CrossSessionEvaluation(
paradigm=paradigm,
datasets=dataset,
suffix="braindecode_example",
overwrite=True,
return_epochs=True,
n_jobs=1,
)
results = evaluation.process(pipes)
print(results.head())
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/analysis/results.py:110: RuntimeWarning: Setting non-standard config type: "MOABB_RESULTS"
set_config("MOABB_RESULTS", mne_data)
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/datasets/bnci/base.py:456: FutureWarning: Montage name 'standard_1005' is deprecated and will be removed in MNE 1.14. Use 'colin27_1005' instead.
montage = make_standard_montage("standard_1005")
epoch train_acc train_loss valid_acc valid_loss dur
------- ----------- ------------ ----------- ------------ ------
1 0.5781 0.6891 0.5172 0.6885 0.2316
2 0.5312 0.7021 0.5172 0.6883 0.2068
3 0.5312 0.6943 0.5172 0.6881 0.2137
4 0.5156 0.6864 0.6552 0.6877 0.2060
5 0.5000 0.6873 0.6552 0.6875 0.2140
6 0.5469 0.6860 0.6207 0.6872 0.2160
7 0.6250 0.6607 0.6207 0.6869 0.2196
8 0.5469 0.6940 0.6207 0.6867 0.2134
9 0.5781 0.6821 0.6207 0.6865 0.2127
10 0.6250 0.6655 0.5862 0.6864 0.2056
epoch train_acc train_loss valid_acc valid_loss dur
------- ----------- ------------ ----------- ------------ ------
1 0.4062 0.7235 0.5517 0.6882 0.2030
2 0.5000 0.7049 0.5517 0.6888 0.2062
3 0.4219 0.7034 0.5517 0.6892 0.2003
Stopping since valid_loss has not improved in the last 3 epochs.
epoch train_acc train_loss valid_acc valid_loss dur
------- ----------- ------------ ----------- ------------ ------
1 0.5469 0.6750 0.3103 0.6995 0.2033
2 0.4531 0.7059 0.3103 0.6971 0.2222
3 0.5625 0.6793 0.3793 0.6957 0.2046
4 0.5625 0.6802 0.4138 0.6950 0.2091
5 0.4375 0.6987 0.4828 0.6943 0.2062
6 0.5312 0.6783 0.4828 0.6939 0.2127
7 0.5625 0.6873 0.4828 0.6936 0.2079
8 0.5156 0.6851 0.4828 0.6935 0.2073
9 0.5625 0.6751 0.4828 0.6935 0.2033
10 0.5938 0.6690 0.4828 0.6934 0.2031
epoch train_acc train_loss valid_acc valid_loss dur
------- ----------- ------------ ----------- ------------ ------
1 0.4844 0.7245 0.7241 0.6872 0.2111
2 0.5000 0.6952 0.6552 0.6877 0.2005
3 0.5469 0.6853 0.6552 0.6881 0.2005
Stopping since valid_loss has not improved in the last 3 epochs.
/opt/hostedtoolcache/Python/3.12.15/x64/lib/python3.12/site-packages/moabb/analysis/results.py:189: H5pyDeprecationWarning: Creating a dataset without passing data or dtype is deprecated. Pass an explicit dtype. Using dtype='f4' will keep the current default behaviour.
dset.create_dataset(
score time samples ... n_sessions dataset pipeline
0 0.488812 3.240729 144.0 ... 2 BNCI2014-001 EEGNet
1 0.537809 0.838966 144.0 ... 2 BNCI2014-001 EEGNet
2 0.470486 2.127158 144.0 ... 2 BNCI2014-001 EEGNet
3 0.472415 0.839005 144.0 ... 2 BNCI2014-001 EEGNet
[4 rows x 11 columns]
Plot Results#
plt.figure()
sns.barplot(data=results, y="score", x="subject", palette="viridis")
plt.show()
/home/runner/work/braindecode/braindecode/examples/advanced_training/plot_moabb_benchmark.py:141: FutureWarning:
Passing `palette` without assigning `hue` is deprecated and will be removed in v0.14.0. Assign the `x` variable to `hue` and set `legend=False` for the same effect.
sns.barplot(data=results, y="score", x="subject", palette="viridis")
References#
Total running time of the script: (0 minutes 15.952 seconds)
Run this example
