1. 版本BUG

最近在模型上线时发现一个 BUG,具体表述为使用 LGBM 2.3.3 加载模型进行预测时,输入特征比训练时少或者多,都可以得到一个预测结果,这就很蛋疼。好在新版本的 LGBM 已经解决了这个问题。

2. 警告信息

某些版本 LGBM 在训练时,会报出以下警告,这并不是 Bug 也不是 Error,只是提示你当前迭代树应该停止,因为不可以再继续分裂了:

  1. [LightGBM] [Warning] No further splits with positive gain, best gain: -inf

解决办法:参数中加上 verbose=-1,就可以消除这些警告信息

  1. params = {
  2. 'boosting': "gbdt",
  3. 'num_leaves': 10,
  4. 'objective': 'binary',
  5. 'max_depth': 4,
  6. 'learning_rate': 0.005,
  7. 'min_gain_to_split': 35,
  8. 'max_bin': 10,
  9. 'max_cat_threshold': 5,
  10. 'bagging_fraction': 0.8,
  11. 'bagging_freq': 10,
  12. 'feature_fraction': [0.8],
  13. 'min_data_in_leaf': [100],
  14. 'seed': [30],
  15. 'init_score': 0.1,
  16. 'metric': ['auc', 'binary_logloss'],
  17. 'verbose': -1
  18. }