1. 版本BUG
最近在模型上线时发现一个 BUG,具体表述为使用 LGBM 2.3.3 加载模型进行预测时,输入特征比训练时少或者多,都可以得到一个预测结果,这就很蛋疼。好在新版本的 LGBM 已经解决了这个问题。
2. 警告信息
某些版本 LGBM 在训练时,会报出以下警告,这并不是 Bug 也不是 Error,只是提示你当前迭代树应该停止,因为不可以再继续分裂了:
[LightGBM] [Warning] No further splits with positive gain, best gain: -inf
解决办法:参数中加上 verbose=-1
,就可以消除这些警告信息
params = {
'boosting': "gbdt",
'num_leaves': 10,
'objective': 'binary',
'max_depth': 4,
'learning_rate': 0.005,
'min_gain_to_split': 35,
'max_bin': 10,
'max_cat_threshold': 5,
'bagging_fraction': 0.8,
'bagging_freq': 10,
'feature_fraction': [0.8],
'min_data_in_leaf': [100],
'seed': [30],
'init_score': 0.1,
'metric': ['auc', 'binary_logloss'],
'verbose': -1
}