更新初始化代码

main.py

@@ -82,7 +82,7 @@ params = {
 }
 
 # 训练模型
-num_round = 1000
+num_round = 100
 bst = lgb.train(params, train_data, num_round, valid_sets=[test_data])
 
 # 预测