Ver código fonte

带标题 tag 去做训练

罗俊辉 1 ano atrás
pai
commit
b3b177e114
1 arquivos alterados com 1 adições e 1 exclusões
  1. 1 1
      main.py

+ 1 - 1
main.py

@@ -145,7 +145,7 @@ class LightGBM(object):
             X_test[key] = pd.to_numeric(X_test[key], errors="coerce")
         bst = lgb.Booster(model_file=self.model)
         y_pred = bst.predict(X_test, num_iteration=bst.best_iteration)
-        y_pred_binary = [0 if i <= 0.1613 else 1 for i in list(y_pred)]
+        y_pred_binary = [0 if i <= 0.15810 else 1 for i in list(y_pred)]
         # 转换为二进制输出
         score_list = []
         for index, item in enumerate(list(y_pred)):