本指南详细介绍如何从Hugging Face平台下载并运用BGE-RERANKER-BASE模型文件,适用于进行文本重排序任务的研究者与开发者。
模型使用:
from sentence_transformers import SentenceTransformer
sentences_1 = [你是谁]
sentences_2 = [你是哪个, 你是谁啊, who are you, 谁]
model = SentenceTransformer(Usershb-macDocumentschatGptllambge-reranker-base)
embeddings_1 = model.encode(sentences_1, normalize_embeddings=True)
embeddings_2 = model.encode(sentences_2, normalize_embeddings=True)
similarity = embeddings_1 @ embeddings_2.T
print(similarity)