使用GPU h2o,我从预测中获得以下输出:
model.predict(data)
False
我正在寻找类似于CPU版本的输出:
model.predict(data)
predict False True
False 0.316711 0.683289
我想查看GPU h2o预测的置信区间。我该怎么办?
答案 0 :(得分:0)
我发现了问题。 h2o4gpu使用与scikit-learn相同的结构。我将预测更改为Forecast_proba(data)并返回了概率