HuggingFaceCrossEncoder#

class langchain_community.cross_encoders.huggingface.HuggingFaceCrossEncoder[source]#

基础类：BaseModel, BaseCrossEncoder

HuggingFace 交叉编码器模型。

示例

from langchain_community.cross_encoders import HuggingFaceCrossEncoder

model_name = "BAAI/bge-reranker-base"
model_kwargs = {'device': 'cpu'}
hf = HuggingFaceCrossEncoder(
    model_name=model_name,
    model_kwargs=model_kwargs
)

初始化 sentence_transformer。

param model_kwargs: Dict[str, Any] [Optional]#

传递给模型的关键字参数。

param model_name: str = 'BAAI/bge-reranker-base'#

使用的模型名称。

score(text_pairs: List[Tuple[str, str]]) → List[float][source]#

使用HuggingFace变压器模型计算相似度分数。

Parameters:

text_pairs (List[Tuple[str, str]]) – 用于评分相似性的文本对列表。

Returns:

分数列表，每个分数对应一对。

Return type:

列表[float]

使用 HuggingFaceCrossEncoder 的示例