MrSimple07 commited on
Commit
c1efa3c
·
1 Parent(s): 8fc7e11

bm = 20, cut off = 0.7

Browse files
Files changed (1) hide show
  1. index_retriever.py +2 -2
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=15
20
  )
21
 
22
  vector_retriever = VectorIndexRetriever(
23
  index=vector_index,
24
  similarity_top_k=30,
25
- similarity_cutoff=0.65
26
  )
27
 
28
  hybrid_retriever = QueryFusionRetriever(
 
16
  try:
17
  bm25_retriever = BM25Retriever.from_defaults(
18
  docstore=vector_index.docstore,
19
+ similarity_top_k=20
20
  )
21
 
22
  vector_retriever = VectorIndexRetriever(
23
  index=vector_index,
24
  similarity_top_k=30,
25
+ similarity_cutoff=0.7
26
  )
27
 
28
  hybrid_retriever = QueryFusionRetriever(