|
a |
|
b/README.md |
|
|
1 |
# EEG-Emotion-classification |
|
|
2 |
|
|
|
3 |
# PROBLEM S TATEMENT |
|
|
4 |
It is difficult to look at the EEG signal and identify the state of Human mind. In this assign- |
|
|
5 |
ment, the SVM classifier is trained with Deap dataset to predict the state of mind. the state of |
|
|
6 |
mind is predicted in terms of valence, arousal. which can further be used to predict the state |
|
|
7 |
of mind in terms of expression. |
|
|
8 |
# PROCEDURE TO SOLVE THE ABOVE PROBLEM |
|
|
9 |
In this assignment, the preprocessed data is used for training the classifier. |
|
|
10 |
Steps involve in training the dataset:- |
|
|
11 |
1. Extracting the dataset |
|
|
12 |
2. Finding the features |
|
|
13 |
3. Reducing the dimension |
|
|
14 |
4. traning the vector |
|
|
15 |
5. checking the classifier efficiency |
|
|
16 |
## EXTRACTING THE DATASET |
|
|
17 |
The DEAP dataset consists of two parts: |
|
|
18 |
1. The ratings from an online self-assessment where 120 one-minute extracts of music |
|
|
19 |
videos were each rated by 14-16 volunteers based on arousal, valence and dominance. |
|
|
20 |
2. The participant ratings, physiological recordings and face video of an experiment where |
|
|
21 |
32 volunteers watched a subset of 40 of the above music videos. EEG and physiological |
|
|
22 |
signals were recorded and each participant also rated the videos as above. |
|
|
23 |
In this assignment, labels are extracted into separate file and data of each channel is extracted |
|
|
24 |
into separate file. data from each channel is stored in row wise versus time in column for each |
|
|
25 |
trail,per person |
|
|
26 |
## FINDING THE FEATURES |
|
|
27 |
In this assignment, Wavelet transform is used to decompose the each channel data into the |
|
|
28 |
five feature i.e |
|
|
29 |
• Delta (< 4 Hz) |
|
|
30 |
• Theta (4-7 Hz) |
|
|
31 |
• Alpha (8-15 Hz) |
|
|
32 |
• Beta (16-31 Hz) |
|
|
33 |
• Gamma (> 32 Hz) |
|
|
34 |
In this assignment, obtained the 7 decomposed values but we negalted the frequency whose |
|
|
35 |
range is in 0-0.5 Hz so that the artifcats are removed. The frequency whose range is near 50 |
|
|
36 |
Hz are removed to reduce the effect of power line on signals. finally, EEG band are obtained |
|
|
37 |
for each channel. |
|
|
38 |
## REDUCING THE DIMENSION |
|
|
39 |
The dimension can be reduced using one of the below mention method:- |
|
|
40 |
1. Standard Deviation |
|
|
41 |
2. Mean |
|
|
42 |
3. Variance |
|
|
43 |
4. Median |
|
|
44 |
But in this assignment Standard Deviation is used because it describe the devaition of each |
|
|
45 |
EEG Band power density properly given by the equation below. |
|
|
46 |
## TRANING THE VECTOR |
|
|
47 |
In this assignment, the classifier used is Support vector machine (SVM). we can also use other |
|
|
48 |
classifier or neural network to predict the values but the training efficiency is found to be |
|
|
49 |
nearly 98 percentage with SVM. |