OpenAI ChatGPT实现简单产品推荐功能python代码.docx
15 浏览量
2023-05-17
11:30:38
上传
评论
收藏 12KB DOCX 举报
OpenAI ChatGPT 实现简单产品推荐功能 python 代码
import openai
import faiss
import pandas as pd
from openai.embeddings_utils import get_embeddings
import numpy as np
#产品搜索
def product_query(product_name):
# 读取商品名称和商品 ID 的数据，并将商品名称转换为文本嵌入向量
df = pd.read_csv("data/products.csv")
product_names = df["product_name"].tolist()
product_ids = df["product_id"].tolist()
embeddings = get_embeddings(product_names,
engine="text-embedding-ada-002")
embeddings = np.array(embeddings).astype('float32')
# 建立 Faiss 向量索引，将商品名称嵌入向量添加到索引中
d = embeddings.shape[1]
index = faiss.IndexFlatL2(d)
index.add(embeddings)
资源评论
中本王
- 粉丝: 168
- 资源: 321
