解读文本实体提取模型的预测结果

请求预测后,Vertex AI 会根据您的模型目标返回结果。实体提取模型的预测为每个文档返回注释,例如检测到的实体的位置、已分配的标签和置信度分数。

置信度可以反映您的模型准确识别每个实体和为其添加标签的置信度。数字越大,模型对预测正确度的置信度越高。

批量预测输出示例

以下示例为经过训练以检测疾病的实体提取模型的预测结果。偏移(开始和结束字符偏移)指定模型在文档中检测到实体的位置,content 字段显示检测到的实体。

显示名会显示模型与每个实体关联的标签,例如 SpecificDiseaseDiseaseClass。标签会按顺序映射到文本段。

{
  "key": 1,
  "predictions": {
    "ids": [
      "1234567890123456789",
      "2234567890123456789",
      "3234567890123456789"
    ],
    "displayNames": [
      "SpecificDisease",
      "DiseaseClass",
      "SpecificDisease"
    ],
    "textSegmentStartOffsets":  [13, 40, 57],
    "textSegmentEndOffsets": [29, 51, 75],
    "confidences": [
      0.99959725141525269,
      0.99912621492484128,
      0.99935531616210938
    ]
  }
}