code for doing ML with eeg data