Logistic Regressions predict_proba returns predictions, not probabilities

h2oai / h2o4gpu

H2Oai GPU Edition

Apache License 2.0

456 stars 96 forks source link

Logistic Regressions predict_proba returns predictions, not probabilities #866

Open henrybm opened 3 years ago

henrybm commented 3 years ago

H2o4gpu's predict_proba() for Logistic Regression is returning [[True True False False ... False True False]]. Sklearn's predict_proba() for Logistic Regression is returning [0.323 0.343 0.284 0.12 ... 0.123 0.543].