f-lauria / eeg_classification

EEG multi-classification task with DL models (CNN, LSTM)

EEG_classification

In this repository you will see how to solve a Multiclass classificationt task in python. Different DL algorithms will be tried.

Tablet of Contents

• General info
• Code
• Technologies
• Dependencies

General info

These data came from a University Deep Learning challenge, which was hosted in Codalab. My team developed different algorithms but here I will show you the ones that I personal created, namely:

• CNN 1D
• CNN 1D + LSTM
• CNN 2D

Moreover, a report can be found here. In the report you can find descritpions of all the models that the team developed, not only the ones shared here.

Code

In this notebook you will find all the models previously mentioned (CNN1D, CNN1D + LSTM and CNN2D).

Technologies

All the algorithms are written in Python. The hardware of my machine is not powerful so I ran all the models in Google Colab, here a nice tutorial about how to use it.

Dependencies

• NumPy
• Pandas
• Sklearn
• Keras