预测代码

main.py

@@ -149,10 +149,10 @@ 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)
-        # # 评估模型
-        # accuracy = accuracy_score(Y_test, y_pred_binary)
-        # print(f'Accuracy: {accuracy}')
+        y_pred_binary = np.where(y_pred > 0.5, 1, 0)
+        # 评估模型
+        accuracy = accuracy_score(Y_test, y_pred_binary)
+        print(f'Accuracy: {accuracy}')
 
 
 if __name__ == '__main__':