gpt4 book ai didi

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

转载 作者:行者123 更新时间:2023-11-30 09:51:14 24 4
gpt4 key购买 nike

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

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

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

最佳答案

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

关于python - 将自定义阈值与 tf.contrib.learn.DNNClassifier 一起使用?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/44856964/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com