MrSimple07 commited on
Commit
a65fa1d
·
1 Parent(s): c22a32c

bm = 30, semantic = 30. hybrid = 60, cutoff = 0.7, topk_reranker= 60

Browse files
Files changed (2) hide show
  1. index_retriever.py +3 -3
  2. utils.py +1 -1
index_retriever.py CHANGED
@@ -16,18 +16,18 @@ def create_query_engine(vector_index):
16
  try:
17
  bm25_retriever = BM25Retriever.from_defaults(
18
  docstore=vector_index.docstore,
19
- similarity_top_k=25
20
  )
21
 
22
  vector_retriever = VectorIndexRetriever(
23
  index=vector_index,
24
- similarity_top_k=25,
25
  similarity_cutoff=0.6
26
  )
27
 
28
  hybrid_retriever = QueryFusionRetriever(
29
  [vector_retriever, bm25_retriever],
30
- similarity_top_k=50,
31
  num_queries=1
32
  )
33
 
 
16
  try:
17
  bm25_retriever = BM25Retriever.from_defaults(
18
  docstore=vector_index.docstore,
19
+ similarity_top_k=30
20
  )
21
 
22
  vector_retriever = VectorIndexRetriever(
23
  index=vector_index,
24
+ similarity_top_k=30,
25
  similarity_cutoff=0.6
26
  )
27
 
28
  hybrid_retriever = QueryFusionRetriever(
29
  [vector_retriever, bm25_retriever],
30
+ similarity_top_k=60,
31
  num_queries=1
32
  )
33
 
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=40,
247
  min_score_threshold=0.5,
248
  diversity_penalty=0.3
249
  )
 
243
  question,
244
  retrieved_nodes,
245
  reranker,
246
+ top_k=60,
247
  min_score_threshold=0.5,
248
  diversity_penalty=0.3
249
  )