coycs commited on
Commit
390bdc0
·
1 Parent(s): 5ae43fb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -265,6 +265,7 @@ def train_model(x, y, test_size, algorithm, paras): # 模型训练
265
  model = XGBRegressor(
266
  max_depth=paras["max_depth"], learning_rate=paras["learning_rate"], n_estimators=paras["n_estimators"])
267
  if algorithm == 5: # LightGBM回归
 
268
  model = lgb.LGBMRegressor(objective='regression', max_depth=paras["max_depth"],
269
  learning_rate=paras["learning_rate"], random_state=paras["random_state"], n_estimators=paras["n_estimators"])
270
 
 
265
  model = XGBRegressor(
266
  max_depth=paras["max_depth"], learning_rate=paras["learning_rate"], n_estimators=paras["n_estimators"])
267
  if algorithm == 5: # LightGBM回归
268
+ # model = lgb.LGBMRegressor(objective='regression',boosting_type="dart",num_leaves=30, max_depth=-1,n_estimators=20,learning_rate=1)
269
  model = lgb.LGBMRegressor(objective='regression', max_depth=paras["max_depth"],
270
  learning_rate=paras["learning_rate"], random_state=paras["random_state"], n_estimators=paras["n_estimators"])
271