|
@@ -27,7 +27,7 @@ params = {
|
|
|
|
|
|
|
|
# 训练模型
|
|
# 训练模型
|
|
|
num_round = 100
|
|
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)
|
|
y_pred = bst.predict(X_test, num_iteration=bst.best_iteration)
|
罗俊辉