FelixLin99 / EEG-Datamining

Construct a model, which contains channel-attention, CNN, LSTM, self-attention, to classify EEG data;
3 stars 0 forks source link
eeg python

🧠 Electroencephalogram (EEG) Data Mining

               -- monitor patients’ real-time attention to track their recovery


" Get to know more about brain signal! "


Created by @Shuhui

Introduction

Environment

Usage

To use model, place the contents of this folder in your PYTHONPATH environment variable.

Use models_of_Felix.py to try CNN-Attention-LSTM-ATTENTION model:

from models_of_Felix import ACRNN_4D

ACRNN_object = ACRNN_4D(input_shape, class_num)
model = ACRNN_object.build_model_withAttention()
model.compile(...)

Use test_EEGTCNET.py to try EEGNET-TCNET-FUSION model

How to preprocess EEG data?

You could find more details in preprocessedPipeline.py

How to do feature extraction?

About our experiment paradigm

Interested in our image generation program? Click here