rerankers 개요 — 통합 리랭커 인터페이스

rerankers 개요 — 통합 리랭커 인터페이스

rerankers는 교차 인코더(cross-encoder), T5 기반, ColBERT 등 다양한 리랭커 모델을 동일한 API로 다루게 해주는 파이썬 라이브러리예요. Python 3.8 이상과 3.9, 3.10, 3.11, 3.12를 지원해요. 모델/방식이 바뀌어도 호출 코드는 거의 그대로예요.

출처: https://github.com/AnswerDotAI/rerankers/blob/main/README.md

기본 사용은 두 줄이면 끝나요.

from rerankers import Reranker

ranker = Reranker("cross-encoder")
results = ranker.rank("What is the tallest building?",
                      ["The Empire State Building", "The Eiffel Tower"])
print(results)
  • Reranker("cross-encoder"): 사용할 리랭커 이름·유형을 지정해요.
  • ranker.rank(query, docs): 쿼리와 문서 목록을 받아 RankedResults를 돌려주고, 각 결과를 .score, .rank, .text로 다룰 수 있어요.

지원 리랭커에는 cross-encoder, T5, ColBERT 계열 등이 있고, 각자 트레이닝 여부와 속도·정확도 트레이드오프가 달라요.

더 알아보기