This commit is contained in:
bohanjason 2019-12-14 23:57:29 -05:00 committed by Dana Van Aken
parent d5f7ae65eb
commit b2e530095e
1 changed files with 0 additions and 2 deletions

View File

@ -482,8 +482,6 @@ def configuration_recommendation(recommendation_input):
workload=mapped_workload, workload=mapped_workload,
task_type=PipelineTaskType.RANKED_KNOBS) task_type=PipelineTaskType.RANKED_KNOBS)
ranked_knobs = JSONUtil.loads(ranked_knobs.data)[:IMPORTANT_KNOB_NUMBER] ranked_knobs = JSONUtil.loads(ranked_knobs.data)[:IMPORTANT_KNOB_NUMBER]
# Sort the important knobs to fix the columns of input X
ranked_knobs = sorted(ranked_knobs)
ranked_knob_idxs = [i for i, cl in enumerate(X_columnlabels) if cl in ranked_knobs] ranked_knob_idxs = [i for i, cl in enumerate(X_columnlabels) if cl in ranked_knobs]
X_workload = X_workload[:, ranked_knob_idxs] X_workload = X_workload[:, ranked_knob_idxs]
X_target = X_target[:, ranked_knob_idxs] X_target = X_target[:, ranked_knob_idxs]