Skip to content

The repository contains trials that apply model transformer to the emotion recognition (classification) task based on electroencephalography(EEG).

Notifications You must be signed in to change notification settings

JoeBao22/eeg_seed_test

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

20 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

EEG SEED TEST

The repository contains trials that apply model transformer to the emotion recognition (classification) task based on electroencephalography(EEG).

experiment 1: transformer for DE feature

dataset

We use the differential entropy(DE) feature from SEED Dataset for the first phase of test. In the data preprocessing part, we generate a feature array(feature.npy) and its corresponding label array(label.npy) for each pair of (chosen feature, average method). The splitting points are written to cumulative array (cumulative.npy).

The Encoder part of transformer is used for classification. We remove the word embedding layer, because our input is already 310 * samples_per_input. To do: We'll adjust the hyper-parameters of transformer for better classification results, and we'll also modify the data_util part, so that the task is subject-dependent.

About

The repository contains trials that apply model transformer to the emotion recognition (classification) task based on electroencephalography(EEG).

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages