eegnetworks 0.1.0

基于PyTorch的各种用于EEG分类任务的简单网络的实现。

EEG-PyTorch

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

已实现网络（Implemented networks）

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

使用方法（Usage）

准备工作

下载数据集 GIB-UVa ERP-BCI dataset

安装依赖包

pip install -r requirements.txt

训练

python train.py --model inception

python train.py --model lstm