1. 学ぶ
  2. /
  3. コース
  4. /
  5. OpenAI API ではじめる Embeddings 入門

Connected

演習

レビューの感情を分類する

すでに埋め込みを計算したので、次はコサイン距離を求めて、最も近いラベルを取り出します。

これを行うために、1つのベクトルと複数のベクトルの埋め込みを比較し、最小距離とそのインデックスを返すfind_closest()という関数を定義します。続いて、レビューをループして各レビューについてfind_closest()で最小距離を見つけ、インデックスを使って分類されたラベルを抽出します。

前の演習で作成したclass_embeddingsとreview_embeddingsのオブジェクトに加えて、reviewsとsentimentsも利用できます。

指示

100 XP
  • find_closest()という関数を定義し、query_vectorに最も類似する埋め込みの距離とインデックスを返してください。
  • 各レビューの埋め込みとclass_embeddingsとの最小距離をfind_closest()で見つけてください。
  • closestの'index'を使ってsentimentsを部分抽出し、'label'を取り出してください。