fix a bug in clean_knob_data()

This commit is contained in:
yangdsh 2020-04-13 05:43:09 +00:00 committed by Dana Van Aken
parent 9fe6ded36c
commit 147ef58065
1 changed files with 1 additions and 1 deletions

View File

@ -127,7 +127,7 @@ def clean_knob_data(knob_matrix, knob_labels, session):
new_lab = knob_name new_lab = knob_name
else: else:
index = knob_labels.index(knob_name) index = knob_labels.index(knob_name)
new_col = knob_matrix[:, index] new_col = knob_matrix[:, index].reshape(-1, 1)
new_lab = knob_labels[index] new_lab = knob_labels[index]
new_labels.append(new_lab) new_labels.append(new_lab)