Metadata-Version: 2.1
Name: EEGNetworks
Version: 0.1.0
Summary:      基于PyTorch的各种用于EEG分类任务的简单网络的实现。     
Home-page: https://github.com/zrr1999/eeg-networks-pytorch
Author: 六个骨头
Author-email: 2742392377@qq.com
License: UNKNOWN
Platform: any
Classifier: Programming Language :: Python :: 3
Classifier: License :: OSI Approved :: GNU Lesser General Public License v3 (LGPLv3)
Classifier: Operating System :: OS Independent
Description-Content-Type: text/markdown

# EEG-PyTorch

基于PyTorch的各种用于EEG分类任务的简单网络的实现。
Implementation of various simple networks based on PyTorch to classify EEG signals.

## 已实现网络（Implemented networks）
- [x] eeg-inception
- [x] eeg-lstm
- [ ] eeg-gcn

## 使用方法（Usage）
### 准备工作

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



安装依赖包

```sh
pip install -r requirements.txt
```

### 训练

```sh
python train.py --model inception
```

```sh
python train.py --model lstm
```

