Emotion-EEG
Set of models for emotion estimation from EEG. Composed by the combination of two deep-learing models learning together (RNN and CNN) with the help of a saliency analysis.
Instruction
The three proposed models are direcly available here:
- Saliency based combination of the hierarchical RNN and CNN. Sal_Model.py
- Feature fusion of the hierarchical RNN and CNN. Feat_Model.pySal_Model* Ouput fusion of the hierarchical RNN and CNN. Loss_Model.py
Installation and Dependencies
Installation with pip: pip install -r req.txt
Import of the environment with conda: conda env create -f env.yml
Remarks
Due to the EULA for each dataset, some example signals have been proposed to test the models, however, they are not corresponding to signals from one of tested dataset.
If you are interested in our work, don't hesitate to contact us.
Best!