Getting probability from DNNclassifier

tensorflow / skflow

Simplified interface for TensorFlow (mimicking Scikit Learn) for Deep Learning

Apache License 2.0

3.18k stars 441 forks source link

Getting probability from DNNclassifier #147

Closed victoyip closed 8 years ago

victoyip commented 8 years ago

In addition to getting the predicted label, I would like to access the probability associated with each prediction. Can I do that with this classifier?

terrytangyuan commented 8 years ago

Use pred_proba instead of predict.

victoyip commented 8 years ago

It said DNNClassifer has no such attribute?

AttributeError: 'TensorFlowDNNClassifier' object has no attribute 'pred_proba'

terrytangyuan commented 8 years ago

Sorry it should be predict_proba.