data-to-loacl

main_spider.py

@@ -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):