Argument parsing with structured class.
Project description
ReRanker
ReRanker for hybrid retrieval.
Installation
pip install reranker
Usage
from datetime import datetime, timedelta
from reranker import ReRanker, CrossEncoderClient, TimeDecayRanker
reranker = ReRanker(
steps=[
CrossEncoderClient(
model_name="cross-encoder/ms-marco-MiniLM-L-6-v2",
addr="http://localhost:8000",
),
TimeDecayRanker(decay_rate=0.5),
]
)
reranker.rank(
query="What is the capital of France?",
docs=[
"Paris is the capital of France.",
"The Eiffel Tower is in Paris.",
"The Louvre is in Paris.",
],
)
reranker.rank(
query=Record(
text="What is the capital of France?",
timestamp=datetime.now(),
),
docs=[
Record(
text="Paris is the capital of France.",
timestamp=datetime.now() - timedelta(days=1),
),
Record(
text="The Eiffel Tower is in Paris.",
timestamp=datetime.now() - timedelta(days=2),
),
Record(
text="The Louvre is in Paris.",
timestamp=datetime.now() - timedelta(days=3),
),
]
)
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
reranker-0.1.0.tar.gz
(9.6 kB
view hashes)