TensorFlow Estimator 1.3无法获得predict_proba?

时间:2017-08-07 21:14:58

标签: python tensorflow

我在1.3(tf.estimator.DNNClassifier)中看到了新的DNN和估算器类,但我没有看到任何获取/设置predict_proba的方法..

因此,截至目前,预测被设置为真实超过.5%并且如果低于二进制情况我会猜错..但是有很多用例预测真实的概率较低,非常有用,特别是非常不平衡的数据?

任何建议或线索?或指导?

BTW可以在这里找到: class tf.contrib.learn.DNNClassifier

1 个答案:

答案 0 :(得分:1)

GitHub上的每个Google:您现在可以使用tf.estimator预测(...,predict_keys =“概率”)

https://github.com/tensorflow/tensorflow/issues/12119#issuecomment-321098690