Esempio n. 1
0
#            optimizer='adam', batch_size=32, loss='categorical_crossentropy',
#            metrics=['accuracy'], test_split_size=0.3, verbose=1)
# print("Done!!!!!!!!!!!!!!!!!")

# channels all
X = np.load('EEG.npy')
####

####
X = reshape_1D_conv(X)
CNN.CNN1D(X,
          y,
          epochs=50,
          name='TimeDomain-allch',
          num_GPU=num_GPU,
          optimizer='adam',
          batch_size=32,
          loss='categorical_crossentropy',
          metrics=['accuracy'],
          test_split_size=0.1,
          verbose=verbose)
'''
# channels 7:21
#X = np.load('EEG.npy')
X = reshape_1D_conv(X[:, :, 7:21])
CNN.CNN1D(X, y, epochs=epochs, name='TimeDomain-ch0:22', num_GPU=num_GPU,
          optimizer='adam', batch_size=32, loss='categorical_crossentropy',
          metrics=['accuracy'], test_split_size=0.1, verbose=verbose)

print("Short Time Fourier 1D")
# Short Time Fourier 1D