diff --git a/example/retriever/retrieval_launch_ann.sh b/example/retriever/retrieval_launch_ann.sh new file mode 100644 index 0000000..f7dc3e7 --- /dev/null +++ b/example/retriever/retrieval_launch_ann.sh @@ -0,0 +1,12 @@ + +file_path=/the/path/you/save/corpus +index_file=$file_path/e5_HNSW64.index +corpus_file=$file_path/wiki-18.jsonl +retriever_name=e5 +retriever_path=intfloat/e5-base-v2 + +python search_r1/search/retrieval_server.py --index_path $index_file \ + --corpus_path $corpus_file \ + --topk 3 \ + --retriever_name $retriever_name \ + --retriever_model $retriever_path diff --git a/example/retriever/retrieval_launch_bm25.sh b/example/retriever/retrieval_launch_bm25.sh new file mode 100644 index 0000000..6c4e1dc --- /dev/null +++ b/example/retriever/retrieval_launch_bm25.sh @@ -0,0 +1,10 @@ + +file_path=/the/path/you/save/corpus +index_file=$file_path/bm25 +corpus_file=$file_path/wiki-18.jsonl +retriever_name=bm25 + +python search_r1/search/retrieval_server.py --index_path $index_file \ + --corpus_path $corpus_file \ + --topk 3 \ + --retriever_name $retriever_name