Skip to main content
支持向量机（SVM） 是一组用于分类、回归和异常值检测的监督学习方法。
本 notebook 介绍如何使用底层基于 scikit-learn 包中 SVM 的检索器。 主要参考自 github.com/karpathy/randomfun/blob/master/knn_vs_svm.html 
pip install -qU  scikit-learn

pip install -qU  lark
我们希望使用 OpenAIEmbeddings，因此需要获取 OpenAI API 密钥。 
import getpass
import os

if "OPENAI_API_KEY" not in os.environ:
    os.environ["OPENAI_API_KEY"] = getpass.getpass("OpenAI API Key:")

OpenAI API Key: ········

from langchain_community.retrievers import SVMRetriever
from langchain_openai import OpenAIEmbeddings

​
使用文本创建新检索器

retriever = SVMRetriever.from_texts(
    ["foo", "bar", "world", "hello", "foo bar"], OpenAIEmbeddings()
)

​
使用检索器

现在可以使用检索器了！ 
result = retriever.invoke("foo")

result

[Document(page_content='foo', metadata={}),
 Document(page_content='foo bar', metadata={}),
 Document(page_content='hello', metadata={}),
 Document(page_content='world', metadata={})]
在 GitHub 上编辑此页 或 提交 issue。
将这些文档连接到 Claude、VSCode 等，通过 MCP 获取实时解答。