eeyhsong / EEG-Conformer

EEG Transformer 2.0. i. Convolutional Transformer for EEG Decoding. ii. Novel visualization - Class Activation Topography.
GNU General Public License v3.0
389 stars 55 forks source link

train_data.npy in CAT.py #12

Closed Sarah-Xu-08 closed 1 year ago

Sarah-Xu-08 commented 1 year ago

Hi! Thank you for sharing the code!

I have a question about the data you use in CAT visualization. There is one line data = np.load('./grad_cam/train_data.npy') but I couldn't find the corresponding file. Could you explain what this file is? What kind of data it contained? Thx!

eeyhsong commented 1 year ago

hello @Sarah-Xu-08, You can directly use your data with shape (batch, feature channel, electrode channel, time samples). Best.

Sarah-Xu-08 commented 1 year ago

Thank you so much for the reply! It solved my problem!