Skip to content
This repository has been archived by the owner on Aug 29, 2023. It is now read-only.
/ EEGNet Public archive

[Old version] PyTorch implementation of EEGNet: A Compact Convolutional Network for EEG-based Brain-Computer Interfaces - https://arxiv.org/pdf/1611.08024.pdf

License

Notifications You must be signed in to change notification settings

aliasvishnu/EEGNet

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

12 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Original authors have uploaded their code here https://github.com/vlawhern/arl-eegmodels

EEGNet

PyTorch implementation of EEGNet: A Compact Convolutional Network for EEG-based Brain-Computer Interfaces

Requirements

  • Python 2
  • Dataset of your own choice, works well with BCI Competition 3 Dataset 2.
  • Pytorch 0.2+
  • Jupyter notebook

Usage

  • GPU - Just shift+enter everything.
  • No GPU - Remove all .cuda(0) before running.

Notes

  • I found ELU to work inferior, would not recommend. Linear units work better than ReLU as well.
  • I found that ELU/Linear/ReLU are similar in performance.

Results

  • BCI Competition 3 Dataset 2 - Fmeasure (0.402)

Credits

Hope this helped you. Raise an issue if you spot errors or contact [email protected].

Releases

No releases published

Packages

No packages published