学习xgboost github地址:https://github.com/dmlc/xgboost/tree/master/demo
import xgboost as xgb
import numpy as np
import pandas as pd
np.set_printoptions(threshold=np.inf) # 将数组数据全部输出
# read in data
dtrain = xgb.DMatrix('demo/data/agaricus.txt.train')
# print(dtrain)
dtest = xgb.DMatrix('demo/data/agaricus.txt.test')
# print(dtest)
# 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)
利用的数据为github里面的data的训练集和测试集
输出:
[0.28583017 0.9239239 0.28583017 0.28583017 0.05169873 0.9239239
0.9239239 0.28583017 0.9239239 0.01072847 0.9239239 0.28583017
0.05169873 0.28583017 0.05169873 0.01072847 0.28583017 0.9239239
0.05169873 0.28583017 0.28583017 0.01072847 0.28583017 0.28583017
0.01072847 0.9239239 0.9239239 0.28583017 0.28583017 0.05169873
0.28583017 0.28583017 0.01072847 0.05169873 0.28583017 0.28583017
0.9239239 0.9239239 0.28583017 0.28583017 0.05169873 0.28583017
0.28583017 0.28583017 0.28583017 0.9239239 0.01072847 0.9239239
0.05169873 0.28583017 0.01072847 0.28583017 0.9239239 0.28583017
0.28583017 0.01072847 0.28583017 0.28583017 0.28583017 0.28583017
0.05169873 0.05169873 0.9239239 0.05169873 0.9239239 0.28583017
0.9239239 0.28583017 0.28583017 0.28583017 0.01072847 0.9239239
0.28583017 0.05169873 0.05169873 0.28583017 0.01072847 0.05169873
0.28583017 0.05169873 0.01072847 0.05169873 0.28583017 0.05169873
0.28583017 0.28583017 0.9239239 0.9239239 0.28583017 0.01072847
0.01072847 0.28583017 0.28583017 0.01072847 0.28583017 0.28583017
0.28583017 0.28583017 0.05169873 0.05169873 0.28583017 0.9239239
0.01072847 0.9239239 0.01072847 0.01072847 0.28583017 0.28583017
0.28583017 0.9239239 0.05169873 0.01072847 0.28583017 0.28583017
0.05169873 0.9239239 0.28583017 0.9239239 0.05169873 0.9239239
0.28583017 0.28583017 0.01072847 0.28583017 0.05169873 0.05169873
0.28583017 0.05169873 0.28583017 0.01072847 0.28583017 0.28583017
0.01072847 0.05169873 0.05169873 0.05169873 0.28583017 0.28583017
0.05169873 0.9239239 0.28583017 0.28583017 0.05169873 0.28583017
0.01072847 0.28583017 0.05169873 0.28583017 0.9239239 0.01072847
0.9239239 0.01072847 0.9239239 0.28583017 0.28583017 0.9239239
0.01072847 0.05169873 0.28583017 0.28583017 0.9239239 0.9239239
0.28583017 0.01072847 0.28583017 0.9239239 0.9239239 0.28583017
0.28583017 0.9239239 0.9239239 0.01072847 0.9239239 0.9239239
0.01072847 0.9239239 0.9239239 0.28583017 0.05169873 0.28583017
0.05169873 0.28583017 0.9239239 0.05169873 0.01072847 0.28583017
0.05169873 0.28583017 0.9239239 0.05169873 0.05169873 0.9239239
0.9239239 0.05169873 0.9239239 0.9239239 0.05169873 0.05169873
0.05169873 0.28583017 0.05169873 0.9239239 0.05169873 0.05169873
0.05169873 0.05169873 0.28583017 0.9239239 0.05169873 0.01072847
0.28583017 0.05169873 0.05169873 0.01072847 0.28583017 0.05169873
0.28583017 0.05169873 0.05169873 0.9239239 0.05169873 0.9239239
0.05169873 0.05169873 0.05169873 0.05169873 0.05169873 0.9239239
0.28583017 0.05169873 0.05169873 0.05169873 0.05169873 0.05169873
0.05169873 0.01072847 0.05169873 0.05169873 0.9239239 0.28583017
0.05169873 0.05169873 0.05169873 0.05169873 0.05169873 0.05169873
0.9239239 0.05169873 0.9239239 0.05169873 0.05169873 0.05169873
0.05169873 0.05169873 0.05169873 0.05169873 0.05169873 0.05169873
0.05169873 0.9239239 0.05169873 0.05169873 0.28583017 0.9239239
0.05169873 0.9239239 0.05169873 0.05169873 0.05169873 0.05169873
0.05169873 0.05169873 0.05169873 0.9239239 0.05169873 0.05169873
0.05169873 0.05169873 0.05169873 0.05169873 0.9239239 0.01072847
0.05169873 0.05169873 0.28583017 0.05169873 0.05169873 0.05169873
0.05169873 0.05169873 0.05169873 0.05169873 0.9239239 0.05169873
0.05169873 0.9239239 0.05169873 0.05169873 0.9239239 0.05169873
0.05169873 0.05169873 0.9239239 0.05169873 0.05169873 0.05169873
0.05169873 0.9239239 0.05169873 0.05169873 0.05169873 0.05169873
0.05169873 0.05169873 0.05169873 0.9239239 0.9239239 0.05169873
0.05169873 0.05169873 0.05169873 0.05169873 0.05169873 0.05169873
0.28583017 0.05169873 0.05169873 0.05169873 0.05169873 0.05169873
0.05169873 0.05169873 0.01072847 0.05169873 0.05169873 0.05169873
0.05169873 0.05169873 0.05169873 0.05169873 0.05169873 0.05169873
0.05169873 0.28583017 0.05169873 0.9239239 0.05169873 0.9239239
0.05169873 0.05169873 0.05169873 0.28583017 0.05169873 0.05169873
0.05169873 0.05169873 0.05169873 0.05169873 0.9239239 0.05169873
0.05169873 0.05169873 0.9239239 0.05169873 0.05169873 0.05169873
0.05169873 0.05169873 0.05169873 0.05169873 0.05169873 0.05169873
0.05169873 0.05169873 0.05169873 0.05169873 0.9239239 0.05169873
0.05169873 0.05169873 0.05169873 0.9239239 0.05169873 0.05169873
0.05169873 0.05169873 0.05169873 0.05169873 0.05169873 0.05169873
0.9239239 0.05169873 0.05169873 0.05169873 0.05169873 0.05169873
0.05169873 0.05169873 0.05169873 0.05169873 0.05169873 0.05169873
0.05169873 0.05169873 0.05169873 0.05169873 0.05169873 0.9239239
0.05169873 0.05169873 0.05169873 0.05169873 0.05169873 0.9239239
0.05169873 0.05169873 0.05169873 0.05169873 0.05169873 0.05169873
0.05169873 0.05169873 0.05169873 0.05169873 0.05169873 0.05169873
0.05169873 0.05169873 0.05169873 0.05169873 0.05169873 0.05169873
0.05169873 0.05169873 0.05169873 0.05169873 0.05169873 0.05169873
0.05169873 0.05169873 0.05169873 0.05169873 0.05169873 0.05169873
0.05169873 0.05169873 0.9239239 0.05169873 0.05169873 0.05169873
0.05169873 0.05169873 0.05169873 0.05169873 0.05169873 0.05169873
0.05169873 0.05169873 0.05169873 0.05169873 0.9239239 0.05169873
0.05169873 0.05169873 0.05169873 0.05169873 0.9239239 0.05169873
0.05169873 0.05169873 0.05169873 0.05169873 0.05169873 0.05169873
0.05169873 0.05169873 0.05169873 0.05169873 0.05169873 0.05169873
0.05169873 0.9239239 0.05169873 0.05169873 0.9239239 0.05169873
0.05169873 0.05169873 0.05169873 0.05169873 0.9239239 0.05169873
0.05169873 0.05169873 0.9239239 0.05169873 0.05169873 0.05169873
0.05169873 0.05169873 0.05169873 0.05169873 0.05169873 0.05169873
0.05169873 0.05169873 0.05169873 0.05169873 0.05169873 0.05169873
0.05169873 0.05169873 0.05169873 0.05169873 0.05169873 0.05169873
0.05169873 0.05169873 0.05169873 0.05169873 0.05169873 0.05169873
0.05169873 0.05169873 0.05169873 0.05169873 0.05169873 0.05169873
0.05169873 0.05169873 0.05169873 0.05169873 0.05169873 0.05169873
0.05169873 0.05169873 0.05169873 0.05169873 0.05169873 0.05169873
0.05169873 0.05169873 0.05169873 0.05169873 0.05169873 0.05169873
0.05169873 0.05169873 0.9239239 0.05169873 0.05169873 0.05169873
0.05169873 0.05169873 0.05169873 0.05169873 0.05169873 0.05169873
0.05169873 0.05169873 0.05169873 0.05169873 0.05169873 0.05169873
0.05169873 0.9239239 0.05169873 0.05169873 0.05169873 0.05169873
0.05169873 0.05169873 0.05169873 0.05169873 0.05169873 0.05169873
0.05169873 0.05169873 0.05169873 0.05169873 0.05169873 0.05169873
0.05169873 0.9239239 0.05169873 0.05169873 0.05169873 0.05169873
0.05169873 0.05169873 0.05169873 0.9239239 0.05169873 0.05169873
0.9239239 0.05169873 0.9239239 0.9239239 0.05169873 0.9239239
0.9239239 0.9239239 0.05169873 0.9239239 0.05169873 0.05169873
0.05169873 0.9239239 0.9239239 0.05169873 0.05169873 0.05169873
0.05169873 0.9239239 0.05169873 0.05169873 0.05169873 0.05169873
0.9239239 0.05169873 0.05169873 0.9239239 0.05169873 0.9239239
0.05169873 0.05169873 0.05169873 0.05169873 0.9239239 0.05169873
0.9239239 0.05169873 0.9239239 0.05169873 0.9239239 0.05169873
0.05169873 0.9239239 0.05169873 0.05169873 0.05169873 0.05169873
0.05169873 0.05169873 0.9239239 0.05169873 0.9239239 0.05169873
0.05169873 0.9239239 0.05169873 0.9239239 0.05169873 0.05169873
0.05169873 0.05169873 0.9239239 0.05169873 0.05169873 0.9239239
0.9239239 0.9239239 0.05169873 0.05169873 0.9239239 0.9239239
0.05169873 0.9239239 0.05169873 0.05169873 0.9239239 0.05169873
0.05169873 0.9239239 0.05169873 0.05169873 0.05169873 0.05169873
0.9239239 0.05169873 0.05169873 0.05169873 0.9239239 0.05169873
0.9239239 0.05169873 0.9239239 0.05169873 0.05169873 0.9239239
0.05169873 0.9239239 0.05169873 0.05169873 0.05169873 0.05169873
0.05169873 0.05169873 0.05169873 0.05169873 0.9239239 0.05169873
0.9239239 0.05169873 0.05169873 0.9239239 0.05169873 0.9239239
0.05169873 0.05169873 0.05169873 0.05169873 0.9239239 0.05169873
0.05169873 0.05169873 0.9239239 0.9239239 0.05169873 0.05169873
0.05169873 0.05169873 0.9239239 0.05169873 0.05169873 0.9239239
0.9239239 0.05169873 0.05169873 0.05169873 0.05169873 0.05169873
0.05169873 0.05169873 0.9239239 0.05169873 0.05169873 0.05169873
0.05169873 0.05169873 0.05169873 0.05169873 0.05169873 0.9239239
0.9239239 0.9239239 0.05169873 0.05169873 0.05169873 0.05169873
0.05169873 0.05169873 0.05169873 0.05169873 0.05169873 0.05169873
0.9239239 0.9239239 0.9239239 0.9239239 0.05169873 0.9239239
0.05169873 0.9239239 0.05169873 0.9239239 0.9239239 0.05169873
0.05169873 0.9239239 0.05169873 0.9239239 0.9239239 0.05169873
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.05169873
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.05169873 0.9239239 0.9239239 0.9239239 0.05169873 0.9239239
0.05169873 0.05169873 0.9239239 0.9239239 0.9239239 0.9239239
0.05169873 0.9239239 0.9239239 0.05169873 0.05169873 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.05169873
0.9239239 0.9239239 0.9239239 0.05169873 0.05169873 0.05169873
0.05169873 0.9239239 0.05169873 0.9239239 0.05169873 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.05169873 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.05169873 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.05169873 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.05169873 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.05169873 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.05169873 0.9239239 0.9239239 0.9239239
0.05169873 0.9239239 0.709121 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.05169873
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.05169873 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.05169873 0.9239239 0.9239239 0.9239239 0.05169873
0.9239239 0.9239239 0.9239239 0.05169873 0.9239239 0.9239239
0.9239239 0.9239239 0.05169873 0.9239239 0.05169873 0.9239239
0.05169873 0.05169873 0.9239239 0.9239239 0.05169873 0.05169873
0.9239239 0.9239239 0.9239239 0.9239239 0.709121 0.05169873
0.05169873 0.9239239 0.05169873 0.05169873 0.9239239 0.05169873
0.05169873 0.9239239 0.9239239 0.9239239 0.709121 0.9239239
0.05169873 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.05169873 0.05169873 0.9239239 0.05169873 0.9239239
0.709121 0.9239239 0.9239239 0.9239239 0.9239239 0.05169873
0.05169873 0.05169873 0.05169873 0.05169873 0.05169873 0.05169873
0.05169873 0.9239239 0.709121 0.05169873 0.9239239 0.9239239
0.9239239 0.05169873 0.9239239 0.9239239 0.9239239 0.9239239
0.05169873 0.9239239 0.9239239 0.05169873 0.9239239 0.709121
0.709121 0.9239239 0.9239239 0.9239239 0.9239239 0.05169873
0.05169873 0.9239239 0.9239239 0.05169873 0.709121 0.9239239
0.05169873 0.9239239 0.9239239 0.05169873 0.9239239 0.9239239
0.9239239 0.05169873 0.709121 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.05169873 0.9239239
0.9239239 0.9239239 0.9239239 0.05169873 0.9239239 0.9239239
0.05169873 0.9239239 0.05169873 0.9239239 0.05169873 0.05169873
0.9239239 0.9239239 0.9239239 0.05169873 0.05169873 0.9239239
0.9239239 0.9239239 0.05169873 0.05169873 0.9239239 0.05169873
0.9239239 0.709121 0.709121 0.9239239 0.05169873 0.9239239
0.05169873 0.05169873 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.709121 0.9239239 0.05169873 0.05169873 0.9239239
0.05169873 0.05169873 0.709121 0.709121 0.9239239 0.9239239
0.9239239 0.709121 0.9239239 0.05169873 0.9239239 0.9239239
0.709121 0.709121 0.9239239 0.9239239 0.05169873 0.709121
0.05169873 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.05169873 0.05169873 0.9239239 0.9239239 0.9239239 0.9239239
0.709121 0.05169873 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.05169873 0.9239239 0.9239239 0.9239239 0.9239239 0.05169873
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.05169873 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.05169873 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.05169873 0.9239239 0.9239239
0.9239239 0.9239239 0.05169873 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.05169873 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.05169873
0.05169873 0.05169873 0.9239239 0.9239239 0.05169873 0.9239239
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.05169873 0.9239239 0.9239239 0.9239239 0.9239239 0.05169873
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.05169873
0.05169873 0.9239239 0.05169873 0.9239239 0.9239239 0.05169873
0.05169873 0.9239239 0.9239239 0.05169873 0.9239239 0.05169873
0.05169873 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.05169873 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.05169873 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.9239239 0.05169873 0.9239239 0.9239239 0.05169873 0.9239239
0.9239239 0.05169873 0.9239239 0.05169873 0.9239239 0.05169873
0.9239239 0.9239239 0.9239239 0.05169873 0.9239239 0.9239239
0.9239239 0.05169873 0.28583017 0.05169873 0.05169873 0.9239239
0.9239239 0.05169873 0.9239239 0.9239239 0.9239239 0.05169873
0.9239239 0.9239239 0.05169873 0.9239239 0.9239239 0.9239239
0.9239239 0.9239239 0.05169873 0.05169873 0.9239239 0.9239239
0.05169873 0.9239239 0.05169873 0.9239239 0.9239239 0.05169873
0.9239239 0.05169873 0.28583017 0.9239239 0.05169873 0.05169873
0.05169873 0.9239239 0.9239239 0.05169873 0.9239239 0.05169873
0.9239239 0.9239239 0.9239239 0.9239239 0.9239239 0.9239239
0.28583017 0.05169873 0.9239239 0.9239239 0.9239239 0.9239239
0.05169873 0.9239239 0.9239239 0.9239239 0.05169873 0.9239239
0.9239239 0.05169873 0.9239239 0.05169873 0.05169873 0.9239239
0.9239239 0.05169873 0.9239239 0.9239239 0.9239239 0.05169873
0.05169873 0.9239239 0.05169873 0.9239239 0.05169873 0.05169873
0.9239239 0.05169873 0.9239239 0.9239239 0.9239239 0.9239239
0.05169873 0.9239239 0.05169873 0.05169873 0.9239239 0.9239239
0.9239239 0.9239239 0.05169873 0.05169873 0.05169873 0.05169873
0.9239239 0.9239239 0.9239239 0.9239239 0.05169873 0.05169873
0.9239239 0.9239239 0.05169873 0.05169873 0.05169873 0.9239239
0.9239239 0.9239239 0.05169873 0.05169873 0.05169873 0.05169873
0.9239239 0.9239239 0.05169873 0.9239239 0.9239239 0.9239239
0.05169873 0.9239239 0.05169873 0.05169873 0.28583017 0.05169873
0.05169873 0.05169873 0.05169873 0.9239239 0.05169873 0.9239239
0.05169873 0.9239239 0.05169873 0.9239239 0.05169873 0.9239239
0.9239239 0.9239239 0.05169873 0.9239239 0.9239239 0.9239239
0.05169873 0.9239239 0.9239239 0.05169873 0.9239239 0.05169873
0.05169873 0.05169873 0.05169873 0.9239239 0.05169873 0.05169873
0.9239239 0.05169873 0.05169873 0.05169873 0.9239239 0.9239239
0.9239239 0.05169873 0.9239239 0.9239239 0.05169873 0.05169873
0.9239239 0.05169873 0.9239239 0.9239239 0.05169873 0.05169873
0.05169873 0.9239239 0.05169873 0.05169873 0.05169873 0.05169873
0.9239239 0.05169873 0.9239239 ]
更多数据集可以参照论文:XGBoost:A Scalable Tree Boosting System