To receive notifications about scheduled maintenance, please subscribe to the mailing-list gitlab-operations@sympa.ethz.ch. You can subscribe to the mailing-list at https://sympa.ethz.ch

Commit f090705f authored by Ard Kastrati's avatar Ard Kastrati
Browse files

Added transpose back because the downsampled data has different structure

parent 2595cdf6
......@@ -22,6 +22,7 @@ def get_mat_data(data_dir, verbose=True):
if verbose: logging.info("Setting the shapes")
X = np.transpose(X, (2, 1, 0))
y = np.transpose(y, (1, 0))
if config['downsampled']: X = np.transpose(X, (0, 2, 1))
if config['model'] == 'eegnet' or config['model'] == 'eegnet_cluster':
X = np.transpose(X, (0, 2, 1))
if verbose:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment