python - 如何在tensorflow中为DNNClassifier使用predict_proba？

如何在tensorflow 1.5中的predict_proba中使用DNNClassifier？我正在使用下面的代码。我相信，与先前的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/"))

之后，当我使用
    predictions =列表（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("Model/"))

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

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

即设置Forecast_keys ='probalities'