python - 将自定义阈值与 tf.contrib.learn.DNNClassifier 一起使用？

Question

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

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

任何帮助将不胜感激，谢谢。

Answer 1

tf.contrib.learn.DNNClassifier 类有一个名为 predict_proba 的方法，它返回给定输入属于每个类的概率。然后，您可以使用诸如 tf.round(prob+thres) 之类的内容通过自定义参数 thres 进行二进制阈值处理。