MrSimple07 commited on
Commit
dbc2de5
·
1 Parent(s): c6d8c50

bm = 40, semantic = 20. hybrid = 50, cutoff = 0.7, topk_reranker= 40

Browse files
Files changed (2) hide show
  1. index_retriever.py +2 -2
  2. utils.py +1 -1
index_retriever.py CHANGED
@@ -16,13 +16,13 @@ def create_query_engine(vector_index):
16
  try:
17
  bm25_retriever = BM25Retriever.from_defaults(
18
  docstore=vector_index.docstore,
19
- similarity_top_k=45
20
  )
21
 
22
  vector_retriever = VectorIndexRetriever(
23
  index=vector_index,
24
  similarity_top_k=20,
25
- similarity_cutoff=0.7
26
  )
27
 
28
  hybrid_retriever = QueryFusionRetriever(
 
16
  try:
17
  bm25_retriever = BM25Retriever.from_defaults(
18
  docstore=vector_index.docstore,
19
+ similarity_top_k=40
20
  )
21
 
22
  vector_retriever = VectorIndexRetriever(
23
  index=vector_index,
24
  similarity_top_k=20,
25
+ similarity_cutoff=0.6
26
  )
27
 
28
  hybrid_retriever = QueryFusionRetriever(
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=40,
247
  min_score_threshold=0.5,
248
  diversity_penalty=0.3
249
  )