Python 是否对tf.contrib.learn.DNNClassifier使用自定义阈值？

我正在处理一个二进制分类问题，我正在TensorFlow中使用tf.contrib.learn.DNNClassifier类。当仅对两个类调用此估计器时，它使用阈值0.5作为两个类之间的截止值。我想知道是否有办法使用自定义阈值，因为这可能会提高模型的准确性

我在网上到处搜索，显然没有办法做到这一点

---

非常感谢您的帮助。

tf.contrib.learn.DNNClassifier类有一个名为

predict\u proba

的方法，该方法返回属于给定输入的每个类的概率。然后，您可以使用类似于，

tf.round（prob+thres）

的东西，使用自定义参数

thres

进行二进制阈值化

哇，我还没有在任何文档中看到过这一点，我现在就试试。非常感谢。事实上，现在我回到了那个男人的页面，我记得看过它，但我没有想到使用它，就像上面解决了你的问题一样，你能考虑接受答案吗？