Skip to content

Implementation of various simple networks based on PyTorch to classify EEG signals.

Notifications You must be signed in to change notification settings

zrr1999/eeg-networks-pytorch

Repository files navigation

English | 简体中文

EEG-PyTorch

Implementation of various simple networks for EEG classification tasks based on PyTorch.

Implemented Networks

  • eeg-inception
  • eeg-lstm
  • eeg-gcn

Usage

Installation

Install using PYPI.

pip install eeg-networks

If you do not want to install the package, you can also clone this project and install the required packages.

git clone https://github.com/zrr1999/eeg-networks-pytorch
cd eeg-networks-pytorch
pip install -r requirements.txt

Training

You can train the model using the following command.

python train.py --model_name inception --device cpu --dataset_path ./GIB-UVA ERP-BCI.hdf5 --model_path ./weights

Use the following command to get more detailed information.

python train.py --help

Validation

You can validate the model using the following command.

python val.py --model_path ./weights/last.pth --model_name inception

Use the following command to get more detailed information.

python val.py --help

References

GIB-UVa ERP-BCI dataset

About

Implementation of various simple networks based on PyTorch to classify EEG signals.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages