LightGBM(Light Gradient Boosting Machine)是一个基于决策树算法的快速的、分布式的、高性能 gradient boosting(GBDT、GBRT、GBM 或 MART)框架,可被用于排行、分类以及其他许多机器学习任务中。
开源项目地址: https://github.com/Microsoft/LightGBM
LightGBM 是一个梯度 boosting 框架,使用基于学习算法的决策树。它可以说是分布式的,高效的,它有以下优势:
更快的训练效率
低内存使用
更好的准确率
支持并行学习
可处理大规模数据
我电脑的环境描述:
编译过程:
lightgbm 依赖 OpenMP 进行编译,苹果的 Clang 不支持。请使用 GCC / G++ 代替。参考官方给出的安装指南:
在终端依次执行一下命令(这里默认你电脑已经有brew):
brew install cmake
brew install gcc
git clone --recursive https://github.com/Microsoft/LightGBM
cd LightGBM
mkdir build
cd build
cmake -DCMAKE_CXX_COMPILER=g++-8 -DCMAKE_C_COMPILER=gcc-8 ..
make -j
cmake的时候注意你电脑安装好的g++的版本,否则不成功
执行到 96% 的时候会卡一会儿,我这里CPU100%了,如果不希望执行“make -j”命令时对电脑影响太大的话,执行“make -j1”,使用CPU的一核,不会占满CPU。
构建成功后会在上层目录生成两个文件,lib_lightgbm.so lightgbm,一个库文件,一个可执行文件。
这个时候可以松口气了,我们使用import lightgbm验证一下,擦了还是报错,表示还没成功,继续往下看
我们需要在终端进入cd到lightGBM/python-package下
继续
export CXX=g++-8 CC=gcc-8
python setup.py install
到此结束,大功告成!