openai通过Embedding实现中文语义相似性对比计算代码

代码语言:python

所属分类:其他

代码描述:openai通过Embedding实现中文语义相似性对比计算代码,输入一段话,从候选文本中通过嵌入对比,找到相似度最高的一个候选文本,通常用于问答匹配。

代码标签: openai Embedding 中文 语义 相似性 对比 计算 代码

下面为部分代码预览,完整代码请点击下载或在bfwstudio webide中打开

# -*- coding: utf-8 -*
import openai, numpy as np


openai.api_key = "openaikey"

def getembedding(txt):
    resp = openai.Embedding.create(
    input=[txt],
    engine="text-similarity-davinci-001")
    return resp['data'][0]['embedding']
    
resp = openai.Embedding.create(
    input=["吃饭", "关机","关闭电脑"],
    engine="text-similarity-davinci-001")

embedding_a.........完整代码请登录后点击上方下载按钮下载查看

网友评论0