python-3.x - 如何计算 h2o4gpu XGBClassifier 中的置信区间？

Question

使用 GPU h2o，我从预测中得到以下输出：

model.predict(data)

False

我正在寻找类似于输出为 CPU 版本的东西：

model.predict(data)

predict     False   True
False   0.316711    0.683289

我想查看 GPU h2o 预测的置信区间。我需要做什么？

标签： python-3.xh2o4gpu

Answer 1

我弄清楚了这个问题。h2o4gpu 使用与 scikit-learn 相同的结构。我将 predict 更改为 predict_proba(data) 并返回了概率