tensorflow / skflow

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Getting probability from DNNclassifier

victoyip opened this issue · comments

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?

Use pred_proba instead of predict.

It said DNNClassifer has no such attribute?

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

Sorry it should be predict_proba.