Browse Source

更新 label 数据

罗俊辉 1 year ago
parent
commit
a2b463acba
1 changed files with 2 additions and 1 deletions
  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)))