0

With GPU h2o, I get the following output from a prediction:

model.predict(data)

False

I am looking for something similar to the CPU version where the output is:

model.predict(data)

predict     False   True
False   0.316711    0.683289

I want to view the confidence interval of the prediction for GPU h2o. What do I need to do?

GBouffard
  • 1,125
  • 4
  • 11
  • 24

1 Answers1

0

I figured out the issue. h2o4gpu uses the same structure as scikit-learn. i changed predict to predict_proba(data) and probabilities were returned