AutoML VISION Google SingleLabel 分类输出 TopK 结果

问题描述

目前 AutoML Vision API 正在输出带有各自分数

的 SingleLabel

例如：

我用 3 个类训练了模型：

1. A
2. B
3. C

然后当我使用 Test & Use 并上传另一张图片时，我只得到 [当前输出]

1. Class A and 0.988437 / 0.99

有没有办法用 Top_K 类获得这种类型的输出（例如 Top 3 (k=3) ） [期望的输出]

1. Class A and 0.988437 / 0.99
2. Class C and 0.3551 / 0.36
3. Class B and 0.1201 / 0.12

根据他们的分数排序。

提前致谢。

解决方法

单标签分类为每个分类图像分配一个标签，并且只返回一个预测类别。

多标签更适合您的用例，因为它允许为图像分配多个标签。

在用户界面（您似乎正在使用的界面）中，您可以指定您希望自定义模型在您create your dataset 时执行的分类类型。

如果出于任何原因，您希望可以选择在单标签分类中获得所有/k 个预测类别的分数，我建议您提出 Feature Request。