TF-IDF检索器#
本教程概述了如何使用使用scikit-learn的TF-IDF的检索器。
有关TF-IDF详细信息,请参见此博客文章 (opens in a new tab)。
from langchain.retrievers import TFIDFRetriever
# !pip install scikit-learn
使用文本创建新的检索器#
retriever = TFIDFRetriever.from_texts(["foo", "bar", "world", "hello", "foo bar"])
使用检索器#
现在我们可以使用检索器了!
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='world', metadata={})]