浏览代码

优化数据处理代码

罗俊辉 1 年之前
父节点
当前提交
fbd3780767
共有 1 个文件被更改,包括 3 次插入1 次删除
  1. 3 1
      main_spider.py

+ 3 - 1
main_spider.py

@@ -156,7 +156,9 @@ class LightGBM(object):
         fw = open("result/summary_{}.txt".format(dt), "a+", encoding="utf-8")
         path = 'data/predict_data/predict_{}.json'.format(dt)
         x, y = self.read_data(path, yc=6)
-        true_label_df = DataFrame(list(y), columns=['ture_label'])
+        print(type(x))
+        print(type(y))
+        # true_label_df = DataFrame(list(y), columns=['ture_label'])
         bst = lgb.Booster(model_file=self.model)
         y_pred = bst.predict(x, num_iteration=bst.best_iteration)
         pred_label_df = DataFrame(list(y_pred), columns=['pred_label'])