소스 검색

更新 label 数据

罗俊辉 1 년 전
부모
커밋
a2b463acba
1개의 변경된 파일2개의 추가작업 그리고 1개의 파일을 삭제
  1. 2 1
      main_spider.py

+ 2 - 1
main_spider.py

@@ -174,7 +174,8 @@ class LightGBM(object):
             real_label = y[index]
             score = item
             prid_label = y_pred_binary[index]
-            print(real_label, "\t", prid_label, "\t", score)
+            if score > 0.6:
+                print(real_label, "\t", prid_label, "\t", score)
             fw.write("{}\t{}\t{}\n".format(real_label, prid_label, score))
             score_list.append(score)
         print("预测样本总量: {}".format(len(score_list)))