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