소스 검색

测试提交

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

+ 1 - 1
main.py

@@ -27,7 +27,7 @@ params = {
 
 # 训练模型
 num_round = 100
-bst = lgb.train(params, train_data, num_round, valid_sets=[test_data], early_stopping_rounds=10)
+bst = lgb.train(params, train_data, num_round, valid_sets=[test_data])
 
 # 预测
 y_pred = bst.predict(X_test, num_iteration=bst.best_iteration)