To receive notifications about scheduled maintenance, please subscribe to the mailing-list You can subscribe to the mailing-list at

Commit 64f06ddc authored by Lukas Wolf's avatar Lukas Wolf
Browse files

debug syntax error

parent 42152ebf
......@@ -56,7 +56,7 @@ class Ensemble_torch:
x = np.transpose(x, (0, 2, 1)) # (batch_size, samples, channels) to (bs, ch, samples) as torch conv layers want it
X_train, y_train, X_val, y_val, X_test, y_test = generate_split(x, y)
# Remove the subject counter from the labels and extract only angle for direction task
if config['task'] == 'amplitude-task':
if config['task'] == 'amplitude-reg':
y_train = y_train[:, 1:2]
y_val = y_val[:, 1:2] # only extract the amplitude, not the subjectid or angle
y_test = y_test[:, 1:2]
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