Update app.py
Browse files
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 |
|