openAI|如何获取嵌入

本文介绍了OpenAI的get_embedding函数,可用于获取输入文本的嵌入向量。通过Python代码示例展示了如何使用该函数,获取指定文本的嵌入向量,进而用于语义搜索、推荐、聚类分析等应用场景。

获取嵌入

函数 get_embedding 将为我们提供输入文本的嵌入。

import openai

embedding = openai.Embedding.create(
    input="Your text goes here", model="text-embedding-ada-002"
)["data"][0]["embedding"]
len(embedding)
1536
import openai
from tenacity import retry, wait_random_exponential, stop_after_attempt


@retry(wait=wait_random_exponential(min=1, max=20), stop=stop_after_attempt(6))
def get_embedding(text: str, model="text-embedding-ada-002") -> list[float]:
    return openai.Embedding.create(input=[text], model=model)["data"][0]["embedding"]


embedding = get_embedding("Your text goes here", model="text-embedding-ada-002")
print(len(embedding))
1536

此文章由OpenAI开源维基百科原创发布,如若转载请注明出处:https://openai.wiki/get_embeddings.html

(0)
上一篇 2023-02-20 12:12
下一篇 2023-02-20 12:19

相关推荐

  • Prompt|简介

    本指南介绍了如何使用提示与大型语言模型进行交互和指令的基础知识,包括温度和top_p参数的配置,以及基本提示和少量提示等。我们还探讨了如何使用指令、QA格式和示例来构建有效的提示。

    ChatGPT 2023-02-18
    002.4K
  • 如何从 Azure OpenAI 获取嵌入

    本文提供了一个Azure嵌入示例,介绍了如何使用OpenAI API创建一个可用于创建嵌入的部署,以及如何将示例嵌入发送到部署。通过可以快速了解如何在Azure中使用OpenAI API进行嵌入操作。

    ChatGPT 2023-02-20
    001.5K
  • ChatGPT|常见报错问题汇总

    本文介绍了ChatGPT常见的网络错误及其解决方法,包括:网络错误, 请求过多, 解决方法, API 接口, Plus 账号,openAI,1小时内请求过多,模型过载,引擎不存在,发生错误等解决方法。

    ChatGPT 2023-02-21
    0210.7K
  • openAI|以 2D 或 3D 形式可视化嵌入

    本文介绍了如何使用t-SNE算法将高维的语言嵌入降至2维,并将结果可视化成散点图。在这个例子中,我们使用了亚马逊上的食品评论数据集,并将每个评论的打分映射成了散点图上点的颜色。

    2023-02-20
    001.3K
  • ChatGPT|制作清晰有效的Prompt指南

    本文介绍了如何制作清晰有效的ChatGPT提示以推动引人入胜且信息丰富的对话,从ChatGPT的基础知识及其工作原理到制作引人注目的提示和解决常见问题的高级技术的所有内容,本文提供了详细的指南和建议。

    ChatGPT 2023-02-18
    022.8K

发表回复

登录后才能评论
微信