テキスト エンティティ抽出モデルの予測結果を解釈する

予測をリクエストすると、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
    ]
  }
}