search

Altaheri / EEG-ATCNet

Attention temporal convolutional network for EEG-based motor imagery classification
https://ieeexplore.ieee.org/document/9852687
Apache License 2.0
173 stars 22 forks source link

Have you written something regarding using your work on a different dataset ? #11

Closed valentindjangone closed 1 year ago

valentindjangone commented 1 year ago

I'm trying to classify 4 classes, motor movements. I would like to try using your work but I have issues with knowing what I have to use and what I can afford to leave on the side in your code. By any chance, do you have written something that would help guide me through using your work on my dataset ?

Altaheri commented 1 year ago

Change the following line that loads the dataset with your own dataset:

        # Get training and test data
        X_train, _, y_train_onehot, X_test, _, y_test_onehot = get_data(
            data_path, sub, LOSO, isStandard)

Your data should has: #classes = 4, #chans = 22, and #samples = 1125. Or you can change the values of these parameters in the def getModel(model_name): function

valentindjangone commented 1 year ago

Thank you! I will give it a try.