Classification of EEG signals into three categories: normal, interictal, and ictal, using 2D convolution neural network
To run the program:
-
Software Python version 3.8.0 Tensorflow 2.4.1 sklearn 1.0.dev0 to use stratifiedGroupKfold otherwise last release CUDA 11.0
-
Set the experiments path and the data path
"exppath" : "D:/research/cnneeg/experiments/", "datapath" : "D:/research/cnneeg/data/images/",either use linux slash (/) style or windows escaped backslash (\)
-
Un compress the file in the data directory
-
In the source directory $ python main.py
-
To generate other data sets, use the matlab provided code