瀏覽代碼

更新 label 数据

罗俊辉 1 年之前
父節點
當前提交
0a201f4661
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      main_spider.py

+ 1 - 1
main_spider.py

@@ -166,7 +166,7 @@ class LightGBM(object):
         y_pred = bst.predict(x, num_iteration=bst.best_iteration)
         pred_score_df = pd.DataFrame(list(y_pred), columns=['pred_score'])
         temp = sorted(list(y_pred))
-        yuzhi = temp[int(len(temp) * 0.5) - 1]
+        yuzhi = temp[int(len(temp) * 0.8) - 1]
         y_pred_binary = [0 if i <= yuzhi else 1 for i in list(y_pred)]
         pred_label_df = pd.DataFrame(list(y_pred_binary), columns=['pred_label'])
         score_list = []