CoolFace
Apppublic

jyotiradityachavan/Temporal_Emotion_Transition_Modeling

sourceHugging Faceupdated 8mo agoView on Hugging Face
0likes
App README

๐Ÿง  EEG Emotion Recognition System

A comprehensive multi-modal cognitive state recognition system that analyzes EEG, eye tracking, GSR, and facial emotion data to identify patterns in cognitive states and emotional responses using advanced machine learning techniques.

๐ŸŒŸ Key Features

๐Ÿ”ฌ Multi-modal Data Integration

  • โ€”EEG Data: Brainwave analysis (Delta, Theta, Alpha, Beta, Gamma bands)
  • โ€”Eye Tracking: Gaze patterns, pupil dilation, fixation metrics
  • โ€”GSR (Galvanic Skin Response): Emotional arousal measurement
  • โ€”Facial Emotion Analysis: Valence, attention, engagement metrics
  • โ€”I-VT Algorithm: Fixation detection and saccade analysis

๐Ÿค– Advanced Machine Learning Pipeline

  • โ€”Hidden Markov Models (HMM): 5-state cognitive modeling
  • โ€”RNN-HMM Hybrid: Temporal sequence modeling with LSTM networks
  • โ€”Seq2Seq with Attention: Transformer-based sequence prediction
  • โ€”Real-time State Classification: Cognitive state identification

๐Ÿ“ˆ Interactive Visualization

  • โ€”Streamlit Dashboard: Web-based interactive interface
  • โ€”t-SNE Trajectory Plots: 2D visualization of cognitive state evolution
  • โ€”Transition Matrix Analysis: State transition probabilities
  • โ€”Real-time Monitoring: Live cognitive state tracking

๐ŸŽฏ Cognitive State Detection

Identify and classify 5 primary cognitive states with physiological indicators:

StateEmojiDescriptionKey Physiological Indicators
State 0๐ŸŽฏFocusedHigh beta waves, stable gaze, reduced blink rate
State 1๐Ÿ˜•DistractedIncreased theta waves, gaze variability, high saccade rate
State 2๐Ÿ˜ดFatiguedElevated alpha waves, slow pupil response, increased blink duration
State 3๐Ÿค”EngagedModerate beta waves, positive valence, steady attention
State 4๐ŸฅฑBoredReduced physiological responses, gaze disengagement

Installation

bash
pip install -r requirements.txt

## Run full pipeline
cd project python main.py
# After completely run of main.py file then start app.py
streamlit run app.py

## Clone the repository
git clone https://github.com/yourusername/eeg-emotion-recognition.git

# Navigate to project directory
cd eeg-emotion-recognition