我正在研究二进制分类问题，并且我在TensorFlow中使用了tf.contrib.learn.DNNClassifier类。当仅为2个类调用此估计器时，它使用阈值0.5作为2个类之间的截止值。我想知道是否有办法使用自定义阈值，因为这可能会提高模型的准确性。
我在网络上搜索过，显然没有办法做到这一点。
非常感谢任何帮助，谢谢。
答案 0 :(得分：1)
tf.contrib.learn.DNNClassifier类有一个名为predict_proba的方法，它返回属于给定输入的每个类的概率。然后，您可以使用tf.round(prob+thres)之类的内容，使用自定义参数thres进行二进制阈值处理。