rank_llm
rank_llm copied to clipboard
Init ServiceRetriever Class
Pull Request Checklist
ServiceRetriever
class for interacting with Anserini REST API. Currently only supports retrieval from pre-built indexes and k=10 (REST API does not accept k as a parameter currently).
Class variables:
-
host
- dataset name (
dataset
) -
retrieval_mode
(only DATASET is valid at the moment) -
retrieval_method
(e.g., BM25)
from_dataset_with_prebuilt_index()
takes in:
-
request
-
dataset_name
-
retrieval_method
-
k
Anserini API currently does not support passing in k -
host
Defaults tolocalhost::8081
PR Type
- [x] Feature