Skip to content

mattjdavis/eeg_classifier

Repository files navigation

EEG classfier

Overview

The repo is an exploration of ML methods to classify EEG signals with siezures present/absent.

Dataset - CHB-MIT Scalp EEG Database

Artifically balanced dataset

This is the result of a Random Forest model trained on a balanced data set, where siezures and non seizure epochs equally represented. Various statstics (peak-to-peak,variance,mean,zero-crossings) from each channel served as features. Just a quick sanity test to get up and running.

Confusion matrix

Figure 1. Histogram of raw data from each class. Qualitative differences suggest a model should be able to classify.

Confusion matrix

Figure 2. Confusion matrix. The model performs quite well on the balanced data.

About

ML methods to classify EEG signals with siezures present/absent

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published