LightGBM MacOS下安装过程及python使用

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

到此结束,大功告成!

猜你喜欢

转载自blog.csdn.net/qq_15111861/article/details/80767080