如何在h2o4gpu XGBClassifier中计算置信区间?

时间:2019-10-16 21:38:04

标签: python-3.x h2o4gpu

使用GPU h2o,我从预测中获得以下输出:

model.predict(data)

False

我正在寻找类似于CPU版本的输出:

model.predict(data)

predict     False   True
False   0.316711    0.683289

我想查看GPU h2o预测的置信区间。我该怎么办?

1 个答案:

答案 0 :(得分:0)

我发现了问题。 h2o4gpu使用与scikit-learn相同的结构。我将预测更改为Forecast_proba(data)并返回了概率