XGBoost预测输出不是二进制的?

2024-04-28 16:07:45 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图运行一个非常简单的示例,其中XGBoost获取一些数据并进行二进制分类。documentation表示xgboost在使用“binary:logistic”时输出概率

import numpy as np
import xgboost as xgb

data = np.random.rand(7,10)
label = np.random.randint(2,size=7)
#print data
#print label

dtrain = xgb.DMatrix(data, label=label)
param = {'bst:max_depth':2, 'bst:eta':1, 'silent':1, 'objective':'binary:logistic' }
plst = param.items()

bst = xgb.train(plst,dtrain,)

dtest= xgb.DMatrix(np.random.rand(4,10))
ypred = bst.predict(dtest)

print ypred

输出为:

[ 0.31350434  0.31350434  0.31350434  0.31350434]

那么这个输出意味着什么呢?这是否意味着我有31%的机会得到1?

如何将其转换为0,1?

这个question似乎是相关的,但我无法从中得到任何有用的东西。


Tags: importdataasnprandomlabelprintbinary