소스 검색

更新 label 数据

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

+ 1 - 1
main_spider.py

@@ -167,7 +167,7 @@ class LightGBM(object):
         pred_score_df = pd.DataFrame(list(y_pred), columns=['pred_score'])
         pred_score_df = pd.DataFrame(list(y_pred), columns=['pred_score'])
         # temp = sorted(list(y_pred))
         # temp = sorted(list(y_pred))
         # yuzhi = temp[int(len(temp) * 0.9) - 1]
         # yuzhi = temp[int(len(temp) * 0.9) - 1]
-        y_pred_binary = [0 if i <= 0.5 else 1 for i in list(y_pred)]
+        y_pred_binary = [0 if i <= 0.169541 else 1 for i in list(y_pred)]
         pred_label_df = pd.DataFrame(list(y_pred_binary), columns=['pred_label'])
         pred_label_df = pd.DataFrame(list(y_pred_binary), columns=['pred_label'])
         score_list = []
         score_list = []
         for index, item in enumerate(list(y_pred)):
         for index, item in enumerate(list(y_pred)):