解读文本分类模型的预测结果

请求预测后,Vertex AI 会根据您的模型目标返回结果。多标签分类模型的预测结果会为每个文档返回一个或多个标签,并为每个标签返回一个置信度分数。对于单标签分类模型,预测结果仅会为每个文档返回一个标签和置信度分数。

置信度分数传达了模型将每个类列或标签与测试项相关联的强度。该数值越高,模型应用于该项的置信度就越高。您可以决定接受模型的置信度分数为多高。

分数阈值滑块

在 Google Cloud 控制台中,Vertex AI 提供一个滑块,用于调整所有类别或标签或者单个类别或标签的置信度阈值。滑块显示在评估标签页的模型详情页面上。置信度阈值是模型为测试项分配类或标签时必须具有的置信度水平。通过调整阈值,您可以看到模型的精度和召回率变化。较高的阈值通常可以提高精确率,并会降低召回率。

批量预测输出示例

以下示例是多标签分类模型的预测结果。该模型对提交的文档应用了 GreatServiceSuggestionInfoRequest 标签。置信度值会按顺序应用于每个标签。在此示例中,模型预测了 GreatService 作为最相关的标签。

{
  "instance": {"content": "gs://bucket/text.txt", "mimeType": "text/plain"},
  "predictions": [
    {
      "ids": [
        "1234567890123456789",
        "2234567890123456789",
        "3234567890123456789"
      ],
      "displayNames": [
        "GreatService",
        "Suggestion",
        "InfoRequest"
      ],
      "confidences": [
        0.8986392080783844,
        0.81984345316886902,
        0.7722353458404541
      ]
    }
  ]
}