预测代码

main.py

@@ -149,7 +149,7 @@ class LightGBM(object):
         data_series = pd.Series(score_list)
         print("统计 score 信息")
         print(data_series.describe())
-        y_pred_binary = np.where(y_pred > 0.5, 1, 0)
+        y_pred_binary = np.where(y_pred > 0.1613, 1, 0)
         # 评估模型
         accuracy = accuracy_score(Y_test, y_pred_binary)
         print(f'Accuracy: {accuracy}')