Thanks to visit codestin.com
Credit goes to github.com

Skip to content

rupansh/EEG-Classification

Repository files navigation

EEG-Classification

EEG Classification of https://www.kaggle.com/competitions/grasp-and-lift-eeg-detection data

Data

https://www.kaggle.com/competitions/grasp-and-lift-eeg-detection/data

Extract train.zip to data/train folder

Dependencies

pip install -r requirements.txt

Training

Available models: basic, inception, resnet

python train_{model}.py

eg:

python train_inception.py

ML Models are still WIP

Testing on validation set

python validate.py

Change the model and state file as required in validate.py

Credits

Basic Net and Loaders: https://www.kaggle.com/code/banggiangle/cnn-eeg-pytorch

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages