OS X上安装XGBoost

1,991次阅读
没有评论

為了要讓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

admin
版权声明:本站原创文章,由admin2017-08-18发表,共计1149字。
转载提示:除特殊说明外本站文章皆由CC-4.0协议发布,转载请注明出处。
评论(没有评论)