Mac系统安装xgboost

Posted on | In , | |

Abstract：Xgboost是大规模并行boosted tree的工具，本文是关于在Mac系统下如何安装XGBoost。

Xgboost

eXtreme Gradient Boosting

Xgboost是大规模并行boosted tree的工具，它是目前最快最好的开源boosted tree工具包，比常见的工具包快10倍以上。

安装步骤

1.安装homebrew (Mac系统下一个非常优秀的包管理工具)

1
/usr/bin/ruby -e "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/master/install)"

2.安装最新版本的gcc

Mac上是没有gcc和g++的, 默认的是clang(安装XCode之后会有).但是XCode自带的clang是不支持OpenMP的.所以我们要自己安装gcc

1
brew install gcc --without-multilib

3.下载xgboost的git源码

1
2
git clone --recursive https://github.com/dmlc/xgboost  
cd xgboost

4.修改配置文件，用于编译

1
cp make/config.mk ./config.mk

用vim打开config.mk，修改下面两行：

1
2
export CC = gcc-6
export CXX = g++-6

5.编译

1
./build.sh

6.安装python版Xgboost

1
2
cd python-package
sudo python setup.py install

7.打开python解释器进行验证，若无报错则安装成功

1
>>>import xgboost as xgb

使用

1
2
3
4
5
6
7
8
9
10
import xgboost as xgb
# read in data
dtrain = xgb.DMatrix('demo/data/agaricus.txt.train')
dtest = xgb.DMatrix('demo/data/agaricus.txt.test')
# specify parameters via map
param = {'max_depth':2, 'eta':1, 'silent':1, 'objective':'binary:logistic' }
num_round = 2
bst = xgb.train(param, dtrain, num_round)
# make prediction
preds = bst.predict(dtest)

Reference

Xgboost官网

Xgboost

Mac系统安装Xgboost

Thanks!