简介
lightgbm是树模型中模型能力最优异的模型之一,作为boosting系的树模型,其运行速度出奇的快,吸收分箱(直方图)的理念大量减少了内存以及计算分叉的时间,最优点分叉而非所有点分叉让其预测结果也出奇的好。
并且lightgbm支持客制化其objective function以及evaluation function,让模型更贴合数据。lightgbm也支持spark的分布式运算。
虽然lightgbm的开源python版本bug很多,但不失为一个十分好用的模型,很适合作为一个baseline快速的得到结果。如果想要详细的了解lightgbm,请移步lightgbm官网或其他网上优秀的教学博客。
本文只会介绍安装lightgbm的坑,以及一个简单例子方便初学的同学使用。
安装
lightgbm在window上的安装很简单,直接使用python自带的pip安装工具安装即可。
在mac上安装会比较困难,用pip安装会遇到错误
1
2
3OSError: dlopen(/Users/{xxx}/anaconda3/lib/python3.6/site-packages/lightgbm/lib_lightgbm.so, 6): Library not loaded: /usr/local/opt/gcc/lib/gcc/7/libgomp.1.dylib
Referenced from: /Users/{xxx}/anaconda3/lib/python3.6/site-packages/lightgbm/lib_lightgbm.so
Reason: image not found
因此我们可以按照以下步骤安装。
安装homebrew
1$ ruby -e "$(cu