ChatGPT 拓展资料:AI大模型之美 -计算两个向量之间的余弦相似度

ChatGPT 拓展资料:AI大模型之美 -计算两个向量之间的余弦相似度

本文讲解使用openai.embeddings_utils中的cosine_similarity和get_embedding函数。

首先,让我们了解一下这两个函数的作用:

  • cosine_similarity: 计算两个向量之间的余弦相似度。
  • get_embedding: 获取一个单词或短语的嵌入向量表示。
    接下来,我将为你演示如何使用这两个函数:

获取单词或短语的嵌入向量表示:要获取单词或短语的嵌入向量表示,你需要调用get_embedding函数,并传递你要获取嵌入向量的单词或短语作为参数。下面是一个使用get_embedding函数获取单词"cat"的嵌入向量的示例:

import openai
import os
from openai.embeddings_utils import cosine_similarity, get_embedding
embedding =  get_embedding("cat")
print(embedding)

这将打印出表示单词"cat"的嵌入向量。

计算两个向量之间的余弦相似度:要计算两个向量之间的余弦相似度,你需要调用cosine_similarity函数,并传递两个向量作为参数。下面是一个使用cosine_similarity函数计算两个向量之间余弦相似度的示例:

vector1 =  get_embedding("cat")
vector2 =  get_embedding("dog")
similarity =  cosine_similarity(vector1, vector2)
print(similarity)

本文链接:https://my.lmcjl.com/post/7826.html

展开阅读全文

4 评论

留下您的评论.