Mac系统安装xgboost

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

Xgboost官网

Xgboost

Mac系统安装Xgboost

btw:欢迎关注 ~

Github: https://github.com/ScarlettYellow

个人博客:https://scarletthuang.cn/



没有更多推荐了,返回首页