我已经尝试了一段时间来弄清楚如何“关闭”LightGBM。特别是,我想在训练期间抑制 LightGBM 的输出(即对提升步骤的反馈)。
我的模型:
params = {
'objective': 'regression',
'learning_rate' :0.9,
'max_depth' : 1,
'metric': 'mean_squared_error',
'seed': 7,
'boosting_type' : 'gbdt'
}
gbm = lgb.train(params,
lgb_train,
num_boost_round=100000,
valid_sets=lgb_eval,
early_stopping_rounds=100)
我尝试verbose=0
按照文档中的建议添加,但这不起作用。
https://github.com/microsoft/LightGBM/blob/master/docs/Parameters.rst
有谁知道如何在训练期间抑制 LightGBM 输出?