注册 登录
    • 为了保证你在浏览本网站时有着更好的体验,建议使用类似Chrome、Firefox之类的浏览器~~
    • 如果你喜欢本站的内容何不Ctrl+D收藏一下呢,与大家一起分享各种编程知识~
    • 本网站研究机器学习、计算机视觉、模式识别~当然不局限于此,生命在于折腾,何不年轻时多折腾一下

OS X上安装XGBoost

Python admin 87次浏览 0个评论 扫描二维码

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


Deeplearn, 版权所有丨如未注明 , 均为原创丨本网站采用BY-NC-SA协议进行授权 , 转载请注明OS X上安装XGBoost
喜欢 (0)
[xiaocui]
分享 (0)

您必须 登录 才能发表评论!