小编aba*_*arc的帖子

Tensorflow 和 Keras 预测阈值

TF 默认用于将输入图像分类为某个类别的阈值是多少?

例如,假设我有 3 个类0, 1, 2,图像的标签是 one-hot 编码的,如下所示:[1, 0, 0],这意味着该图像的标签为 0 类。

现在,当模型在 softmax 之后输出预测时,如下所示:[0.39, 0.56, 0.05]TF 是否使用 0.5 作为阈值,因此它预测的类别是类别 1?

如果所有预测都低于 0.5,就像[0.33, 0.33, 0.33]TF 所说的结果一样怎么办?

有没有什么方法可以指定一个新的阈值(例如 0.7),并确保 TF 在没有类别预测高于该阈值的情况下表示预测是错误的?

此外,这种逻辑是否也会延续到推理阶段,如果网络不确定该类别,那么它将拒绝对图像进行分类?

classification machine-learning keras tensorflow

5
推荐指数
1
解决办法
8453
查看次数