如何在tensorflow中为DNNClassifier使用predict_proba？

Question

如何在 tensorflow 1.5 中将 predict_proba 用于 DNNClassifier？我正在使用下面的代码。我相信，与以前的 tensorflow 版本相比，获取概率而不是确切结果 class 的命令可能已经发生了变化。

new_classifier = 
tf.estimator.DNNClassifier(feature_columns=feature_columns, hidden_units=
[10, 20, 10],n_classes=int(trn_classes),model_dir=os.path.dirname("Model/"))

之后，当我使用 predictions =列表(new_classifier.predict_proba(input_fn=predict_input_fn))

我收到错误-

AttributeError: 'DNNClassifier' object has no attribute 'predict_proba'

请帮忙

Answer 1

如果您的估算器是：

new_classifier = tf.estimator.DNNClassifier(feature_columns=feature_columns, hidden_units=[10, 20, 10],n_classes=int(trn_classes),model_dir=os.path.dirname("Model/"))

对于每个预测尝试的概率：

y_out_prob=new_classifier.predict(input_fn=predict_input_fn,predict_keys="probabilities")

即设置predict_keys='probalities'

如何在tensorflow中为DNNClassifier使用predict_proba？

How to use predict_proba for DNNClassifier in tensorflow?

python

dotnetnuke

tensorflow