使用 GPU h2o,我从预测中得到以下输出:
model.predict(data)
False
我正在寻找类似于输出的 CPU 版本的东西:
model.predict(data)
predict False True
False 0.316711 0.683289
我想查看 GPU h2o 预测的置信区间。 我需要做什么?
我想通了这个问题。 H2O4GPU使用与Scikit-learn相同的结构。 我将预测更改为 predict_proba(数据(并返回概率