search

zwcolin / EEG-Transformer

A ViT based transformer applied on multi-channel time-series EEG data for motor imagery classification
GNU General Public License v3.0
146 stars 17 forks source link

which dataset do you use #1

Closed h1deOnBush closed 1 year ago

h1deOnBush commented 1 year ago

Hi I just wonder Which dataset from BCI Competition IV are you using?

zwcolin commented 1 year ago

Dataset 1. A code snippet for getting the dataset has been provided in util.py.

h1deOnBush commented 1 year ago

thx! And by the way, is timm a private tool repository ？

zwcolin commented 1 year ago

timm is a python package primarily for cv that includes implementations of different architectures, optimizers, losses, etc. Just do pip install timm and it should be good. Let me know if you have any issues caused by version incompatibility.