EEG Emotion Classification with Transformers (PyTorch)

🚀 Transformer-based classifier for emotional state prediction from EEG signals. Built with PyTorch.

Features

• Transformer encoder for EEG sequence modeling
• Multiclass classification (e.g., happy/sad/neutral)
• Fake EEG dataset included for demo use
• Ready for real EEG input from OpenBCI or other devices

Quickstart

pip install -r requirements.txt
python create_fake_data.py
python train.py

Author

Created by Soroush Zare

License

MIT License