MrSimple07 commited on
Commit
f49e798
·
1 Parent(s): 9e1a00e

top k in reranker = 30, bm = 20

Browse files
Files changed (2) hide show
  1. index_retriever.py +1 -1
  2. utils.py +1 -1
index_retriever.py CHANGED
@@ -16,7 +16,7 @@ def create_query_engine(vector_index):
16
  try:
17
  bm25_retriever = BM25Retriever.from_defaults(
18
  docstore=vector_index.docstore,
19
- similarity_top_k=60
20
  )
21
 
22
  vector_retriever = VectorIndexRetriever(
 
16
  try:
17
  bm25_retriever = BM25Retriever.from_defaults(
18
  docstore=vector_index.docstore,
19
+ similarity_top_k=20
20
  )
21
 
22
  vector_retriever = VectorIndexRetriever(
utils.py CHANGED
@@ -243,7 +243,7 @@ def answer_question(question, query_engine, reranker, current_model, chunks_df=N
243
  question,
244
  retrieved_nodes,
245
  reranker,
246
- top_k=20,
247
  min_score_threshold=0.5,
248
  diversity_penalty=0.3
249
  )
 
243
  question,
244
  retrieved_nodes,
245
  reranker,
246
+ top_k=30,
247
  min_score_threshold=0.5,
248
  diversity_penalty=0.3
249
  )