Question

如何在张量流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'

请帮忙。

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中使用predict_proba作为DNNClassifier？

1 个答案: