MrSimple07 commited on
Commit
fe5790f
·
1 Parent(s): 23c53c9

new top k 30

Browse files
Files changed (1) hide show
  1. index_retriever.py +2 -2
index_retriever.py CHANGED
@@ -22,12 +22,12 @@ def create_query_engine(vector_index):
22
  vector_retriever = VectorIndexRetriever(
23
  index=vector_index,
24
  similarity_top_k=30,
25
- similarity_cutoff=0.8
26
  )
27
 
28
  hybrid_retriever = QueryFusionRetriever(
29
  [vector_retriever, bm25_retriever],
30
- similarity_top_k=30,
31
  num_queries=1
32
  )
33
 
 
22
  vector_retriever = VectorIndexRetriever(
23
  index=vector_index,
24
  similarity_top_k=30,
25
+ similarity_cutoff=0.85
26
  )
27
 
28
  hybrid_retriever = QueryFusionRetriever(
29
  [vector_retriever, bm25_retriever],
30
+ similarity_top_k=15,
31
  num_queries=1
32
  )
33