Skip to content

martinwimpff/eeg-online

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

11 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

eeg-online

Official repository for Tailoring Deep Learning for Real-Time Brain-Computer Interfaces: From Offline Models to Calibration-Free Online Decoding.

Usage

Data

Download the datasets from zenodo(Dreyer) and GigaDB(Lee). Place the data in the data folder.

Installation

Run pip install . to install the eeg-online package.

Note: you can also use poetry for the installation

Source training

Run train_test.py with a --config of your choice.

Supervised and unsupervised finetuning

Run finetune.py with the corresponding --source_config and a --config of your choice.

We included some checkpoints (for the first 5 subjects) in lightning_logs to provide a starting point.

Online test-time adaptation

Run run_online_adaptation.py with the corresponding --source_config and a --config of your choice.

In case of any specific questions please contact me or create an issue.

Citation

If you find this repository useful, please cite our work

@article{202407.2370,
	doi = {10.20944/preprints202407.2370.v2},
	url = {https://doi.org/10.20944/preprints202407.2370.v2},
	year = 2024,
	month = {September},
	publisher = {Preprints},
	author = {Martin Wimpff and Jan Zerfowski and Bin Yang},
	title = {Tailoring Deep Learning for Real-Time Brain-Computer Interfaces: From Offline Models to Calibration-Free Online Decoding},
	journal = {Preprints}
}