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

OS X上安装XGBoost

Python admin 10个月前 (08-18) 413次浏览 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)
admin
关于作者:

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