/channel-attention

Primary LanguagePythonMIT LicenseMIT

EEG motor imagery decoding: A framework for comparative analysis with channel attention mechanisms

This is the official repository to the paper EEG motor imagery decoding: A framework for comparative analysis with channel attention mechanisms.

Usage

Data

All data will be downloaded automatically except for the BCIC III dataset. Download the BCIC III dataset and put all files in the directory defined in load_bcic3.

Installation

  • clone this repository
  • run pip install . to install the channel-attention package

Note: you can also use poetry for the installation

Model training

  • run run.py with the --config of your choice

Citation

If you find this repository useful, please cite our work

@article{wimpff2023eeg,
  title={EEG motor imagery decoding: A framework for comparative analysis with channel attention mechanisms},
  author={Wimpff, Martin and Gizzi, Leonardo and Zerfowski, Jan and Yang, Bin},
  journal={arXiv preprint arXiv:2310.11198},
  year={2023}
}