jaredloomis / neurothink

Muse 2 EEG analysis with PyTorch + CNNs.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

neurothink

neurothink is project exploring Muse 2 EEG data.

EEG 10-10 system

Eyes open/closed detection

notebook

In this notebook, I train a CNN to determine whether the wearer's eyes are open or closed based on the raw EEG signals. The results were surprising, with up to 82% accuracy on my dataset.

Possible improvements:

  • Use FFT data as additional features (ie. feature per band per sample). example
  • Use stationary wavelet transform, as a better (?) alternative to FFT.
  • Experiment with network architectures.

Direction detection

notebook

Here, I train a similar CNN to determine whether the wearer is thinking of the concept of left, right, or neither. In Progress.

Meditation coach

Next, I plan on trying to training a network to detect different states of meditation, preferably with some notion of "meditation quality" baked into the training data and therefore the resulting network outputs.

About

Muse 2 EEG analysis with PyTorch + CNNs.


Languages

Language:Jupyter Notebook 98.6%Language:Python 1.3%Language:Shell 0.1%