OS X上安装XGBoost

共计 1149 个字符，预计需要花费 3 分钟才能阅读完成。

為了要讓XGBoost在mac上具有multi-threading的功能，我們必須另外安裝gcc

首先安裝Homebrew

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

再來安裝gcc (預設位置在/usr/local/Cellar/gcc)可以查看gcc版本

brew install gcc --without-multilib

注意！這裡如果沒有指定brew安裝gcc的版本會預設為最新(目前為7.1.0)

cd到放xgboost的目錄位置，接著clone XGBoost

cd <directory>
git clone --recursive https://github.com/dmlc/xgboost

打開make/config.mk 後uncomment掉底下兩行

export CC = gcc
export CXX = g++

接著修改這兩行為(如果安裝gcc為6.X.X則為-6)

export CC = gcc-7
export CXX = g++-7

儲存後cd到/xgboost 輸入以下命令

cp make/config.mk .
make -j4

最後輸入

cd python-package; sudo python setup.py install

應該就可以在conda list下面看到xgboost了！

安装完简单测试一下

import xgboost as xgb
# read in data
dtrain = xgb.DMatrix('/Users/leiyang/xgboost/demo/data/agaricus.txt.train')
dtest = xgb.DMatrix('/Users/leiyang/xgboost/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)
print preds
输出结果如下
[ 0.28583017 0.92392391 0.28583017 ..., 0.92392391 0.05169873
 0.92392391]

转载自https://medium.com/@morris_tai/%E5%A6%82%E4%BD%95%E5%9C%A8osx%E4%B8%8A%E5%AE%89%E8%A3%9Dxgboost-7133c943e33c