python之如何在tensorflow中为DNNClassifier使用predict_proba

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

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/")) 

在那之后，当我使用 预测= 列表(new_classifier.predict_proba(input_fn=predict_input_fn))

我得到错误-

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

请帮忙。

请您参考如下方法:

如果您的估算器是:

new_classifier = tf.estimator.DNNClassifier(feature_columns=feature_columns, hidden_​​units=[10, 20, 10],n_classes=int(trn_classes),model_dir=os.path.dirname("模型/"))

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

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

即设置predict_keys='probalities'