fix a bug in clean_knob_data()
This commit is contained in:
parent
9fe6ded36c
commit
147ef58065
|
@ -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)
|
||||||
|
|
Loading…
Reference in New Issue