ollama+mxbai-embed-large+FAISS实现语句向量化embedding查询代码

代码语言:python

所属分类:其他

代码描述:ollama+mxbai-embed-large+FAISS实现语句向量化embedding查询代码

代码标签: ollama mxbai-embed-large FAISS 语句 向量化 embedding 查询

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

import ollama
from langchain_community.vectorstores import FAISS
import numpy as np

# Step 1: Generate embeddings using ollama
model = 'mxbai-embed-large'
prompt = '自然语言'

# Generate the embedding for the query sentence
query_embedding = ollama.embeddings(model=model, prompt=prompt)

# Ensure the embedding is a numpy array
query_embedding = np.array(query_embedding['embedding'])  # Access the.........完整代码请登录后点击上方下载按钮下载查看

网友评论0