|
|
@@ -187,7 +187,7 @@ class LightGBM(object):
|
|
|
# 水平合并
|
|
|
df_concatenated = pd.concat([ids, titles, x, true_label_df, pred_score_df, pred_label_df], axis=1)
|
|
|
for key in self.str_columns:
|
|
|
- df_concatenated[key] = self.label_mapping[key][df_concatenated[key]]
|
|
|
+ df_concatenated[key] = [self.label_mapping[key][i] for i in df_concatenated[key]]
|
|
|
df_concatenated.to_excel("data/predict_data/spider_predict_result_{}.xlsx".format(dt), index=False)
|
|
|
|
|
|
def feature_importance(self):
|
罗俊辉