Abstract:Xgboost是大规模并行boosted tree的工具,本文是关于在Mac系统下如何安装XGBoost。
XGBOOST
eXtreme Gradient Boosting
Xgboost是大规模并行boosted tree的工具,它是目前最快最好的开源boosted tree工具包,比常见的工具包快10倍以上。
安装步骤
1.安装homebrew (Mac系统下一个非常优秀的包管理工具)
/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
brew install gcc --without-multilib
3.下载xgboost的git源码
git clone --recursive https://github.com/dmlc/xgboost cd xgboost
4.修改配置文件,用于编译
cp make/config.mk ./config.mk
用vim打开config.mk,修改下面两行:
export CC = gcc-6 export CXX = g++-6
5.编译
./build.sh
6.安装python版Xgboost
cd python-package sudo python setup.py install
7.打开python解释器进行验证,若无报错则安装成功
>>>import xgboost as xgb
使用
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
btw:欢迎关注 ~
Github: https://github.com/ScarlettYellow
个人博客:https://scarletthuang.cn/