linux下xgboost的安装与测试

在运行前,要先将python安装好, gcc 5 以上, cmake版本也不能太低

1 编译安装基础包

git clone --recursive https://github.com/dmlc/xgboost

下载后的档案在xgboost文件夹里

cd xgboost

mkdir build

cd build

cmake .. 

如果报错可以先upgrade gcc,cmake版本

make -j$(nproc)   //nproc是操作系统级别对每个用户创建的进程数的限制make -j4的意思是开辟四个并行

 

2. 用管理员权限安装python接口,在xgboost文件夹里有一个python-package

cd python-package/

python setup.py install

我用的是anaconda.最终安装在这个路径/usr/local/anaconda3/lib/python3.7/site-packages/xgboost

 

3 测试

在xgboost的目录下创建test.py

vi test.py

内容: 

import xgboost as xgb

dtrain = xgb.DMatrix('demo/data/agaricus.txt.train')

dtest = xgb.DMatrix('demo/data/agaricus.txt.test')

param = {'max_depth':2, 'eta':1, 'objective':'binary:logistic' }

num_round = 2

bst = xgb.train(param, dtrain, num_round)

preds = bst.predict(dtest)

print(preds)

退出后,运行python test.py