CODEWITHSUNDEEP
pythontensorflow
dartdog
dartdog

Reputation: 10872

TensorFlow Estimator 1.3 no way to get predict_proba?

I see the new DNN and estimator classes in 1.3 (tf.estimator.DNNClassifier) but I don't see any way to get/set predict_proba..

So as of now a prediction is set to true at over .5 % and false if under for the binary case I guess.. but there are numerous use cases where a lower probability to predict true, is very useful, particularly with very unbalanced data?

Any advice or leads? or guidance?

BTW it is available here: class tf.contrib.learn.DNNClassifier

Upvotes: 3

Views: 1136

Answers (1)

dartdog
dartdog

Reputation: 10872

Per Google at GitHub: You can use tf.estimator now with e.g. predict(..., predict_keys="probabilities")

https://github.com/tensorflow/tensorflow/issues/12119#issuecomment-321098690

Upvotes: 1

Related Questions