This project implements a baseline model for EEG signal classification using:
- MiniROCKET (for time series feature extraction)
- RidgeClassifierCV (for classification)
- BCI Competition IV Dataset 2a (BNCI2014_001) via MOABB.
- Using only Subject 1, Session '0train', Run '0' for this baseline.
- Load raw EEG signals.
- Extract event markers (motor imagery tasks: feet, left hand, right hand, tongue).
- Epoch signals into trials (0–2 seconds windows).
- Transform using MiniROCKET.
- Train RidgeClassifierCV.
- Evaluate accuracy & classification metrics.
Metric | Value |
---|---|
Accuracy | ~50% |
Macro F1 | ~0.47 |
Support | 10 samples (small test set) |
Note: Accuracy is low because only a small subset (run '0' of subject 1) was used. This is intentional to establish a reproducible baseline.
- Combine multiple runs per subject (e.g., runs '1', '2', '3', etc.).
- Include more subjects (1–9) for better generalization.
- Experiment with S-Rocket, HDC-Rocket as enhancements.
- Evaluate on larger test splits & cross-validation.
pip install -r requirements.txt
python src/data_preprocessing.py