MrSimple07 commited on
Commit
8e8dddf
·
1 Parent(s): 8d6a517

bm = 200 + cut off 0.35

Browse files
Files changed (1) hide show
  1. index_retriever.py +2 -2
index_retriever.py CHANGED
@@ -46,13 +46,13 @@ def create_query_engine(vector_index):
46
 
47
  bm25_retriever = BM25Retriever.from_defaults(
48
  docstore=vector_index.docstore,
49
- similarity_top_k=150
50
  )
51
 
52
  vector_retriever = VectorIndexRetriever(
53
  index=vector_index,
54
  similarity_top_k=150,
55
- similarity_cutoff=0.55
56
  )
57
 
58
  hybrid_retriever = QueryFusionRetriever(
 
46
 
47
  bm25_retriever = BM25Retriever.from_defaults(
48
  docstore=vector_index.docstore,
49
+ similarity_top_k=200
50
  )
51
 
52
  vector_retriever = VectorIndexRetriever(
53
  index=vector_index,
54
  similarity_top_k=150,
55
+ similarity_cutoff=0.35
56
  )
57
 
58
  hybrid_retriever = QueryFusionRetriever(