kNN
本笔记本演示了如何使用基于kNN的检索器。
主要基于https://github.com/karpathy/randomfun/blob/master/knn_vs_svm.ipynb (opens in a new tab)
from langchain.retrievers import KNNRetriever
from langchain.embeddings import OpenAIEmbeddings
使用文本创建新的检索器。#
retriever = KNNRetriever.from_texts(["foo", "bar", "world", "hello", "foo bar"], OpenAIEmbeddings())
使用检索器 Use Retriever#
现在我们可以尝试使用我们的检索器了！
result = retriever.get_relevant_documents("foo")
result
[Document(page_content='foo', metadata={}),
Document(page_content='foo bar', metadata={}),
Document(page_content='hello', metadata={}),
Document(page_content='bar', metadata={})]