Getting probability from DNNclassifier
victoyip opened this issue · comments
Victor Yip commented
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?
Yuan Tang commented
Use pred_proba instead of predict.
Victor Yip commented
It said DNNClassifer has no such attribute?
AttributeError: 'TensorFlowDNNClassifier' object has no attribute 'pred_proba'
Yuan Tang commented
Sorry it should be predict_proba.