https://github.com/zrr1999/eeg-networks-pytorch

Implementation of various simple networks based on PyTorch to classify EEG signals.
https://github.com/zrr1999/eeg-networks-pytorch

deep-learning eeg-classification eeg-signals pytorch

Last synced: 3 months ago
JSON representation

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

• Host: GitHub
• URL: https://github.com/zrr1999/eeg-networks-pytorch
• Owner: zrr1999
• Created: 2022-06-01T14:04:59.000Z (over 2 years ago)
• Default Branch: main
• Last Pushed: 2023-12-02T04:37:16.000Z (about 1 year ago)
• Last Synced: 2023-12-03T03:30:06.498Z (about 1 year ago)
• Topics: deep-learning, eeg-classification, eeg-signals, pytorch
• Language: Python
• Homepage:
• Size: 30.3 KB
• Stars: 4
• Watchers: 0
• Forks: 0
• Open Issues: 2
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
English | [简体中文](README.zh-CN.md)

# EEG-PyTorch

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

## Implemented Networks

- [x] eeg-inception

- [x] eeg-lstm

- [ ] eeg-gcn

## Usage

### Installation

Install using PYPI.

```sh

pip install eeg-networks

```

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

```sh

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.

```sh

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.

```sh

python train.py --help

```

### Validation

You can validate the model using the following command.

```sh

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

```

Use the following command to get more detailed information.

```sh

python val.py --help

```

## References

[GIB-UVa ERP-BCI dataset](https://www.kaggle.com/datasets/esantamaria/gibuva-erpbci-dataset?resource=download)