Commit 31727f2f authored by Noric Couderc's avatar Noric Couderc
Browse files

Fixed bug in train_model 

The software_selected_columns and hardware_selected_columns
used to have different types, now they are the same.
parent a1b60cbc

train_model.py

+1 −1
Original line number Diff line number Diff line
@@ -288,7 +288,7 @@ def load_training_data(jmh_results_filename, 
    jmh_with_best["Sample weight"] = compute_sample_weights(jmh_with_best)

    software_data = load_software_counters(software_counters_filename)

    software_with_jmh = merge_jmh_software(jmh_with_best, software_data)

    software_selected_columns = software_data.columns

    software_selected_columns = list(software_data.columns)

    software_features = software_with_jmh.get(software_selected_columns)

    papi_data = load_hardware_counters(hardware_counters_filename)

    papi_data['size'] = jmh_with_best['Param: applicationSize']