A Domain Generative Graph Network for EEG-based Emotion Recognition

This is a pytorch implementation of the paper

Environment

• Pytorch 1.8.1
• Python 3.8.0
• cuda 11.1

Network structure

Fig. 1. The schematic framework of DGGN.

Fig. 2. The Framework of Generator.

Dataset

Download target dataset DEAP and SEED from http://www.eecs.qmul.ac.uk/mmv/datasets/deap and https://bcmi.sjtu.edu.cn/~seed/index.html respectively.

Pretrain

subject dependent experiment: run

python pretrain_and_gan_dep.py

subject independent experiment: run

python pretrain_and_gan_indep.py

Train

subject dependent experiment: run

python train_dep.py

subject independent experiment: run

python train_indep.py