overview 노트북 — 리랭커 기본 사용 실습

overview 노트북 — 리랭커 기본 사용 실습

overview 노트북은 rerankers의 가장 기본적인 사용법을 하나씩 보여줘요. 다양한 리랭커를 만들고, 같은 문서 집합에 대해 점수·순위를 비교하면서 각 리랭커의 특징을 파악할 수 있어요.

출처: https://github.com/AnswerDotAI/rerankers/blob/main/examples/overview.ipynb

from rerankers import Reranker
docs = ["Paris is the capital of France", "Tokyo is the capital of Japan"]
query = "What is the capital of France?"

ranker = Reranker("cross-encoder", model="cross-encoder/ms-marco-MiniLM-L-6-v2")
res = ranker.rank(query=query, docs=docs)
for r in res.results:
    print(r.rank, r.text, r.score)

Reranker에 이름(쇼트컷) 또는 명시적 model= 인자를 주고, 토크나이저·배치 사이즈 같은 인자를 더 줄 수 있어요. res.results는 정렬된 결과 목록이고, .top은 최상위 하나를 줍니다.

더 알아보기