MrSimple07 commited on
Commit
9e1a00e
·
1 Parent(s): edc2f6f

bm = 60, semantic = 50. hybrid = 70

Browse files
Files changed (1) hide show
  1. index_retriever.py +2 -2
index_retriever.py CHANGED
@@ -16,7 +16,7 @@ def create_query_engine(vector_index):
16
  try:
17
  bm25_retriever = BM25Retriever.from_defaults(
18
  docstore=vector_index.docstore,
19
- similarity_top_k=50
20
  )
21
 
22
  vector_retriever = VectorIndexRetriever(
@@ -27,7 +27,7 @@ def create_query_engine(vector_index):
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=60
20
  )
21
 
22
  vector_retriever = VectorIndexRetriever(
 
27
 
28
  hybrid_retriever = QueryFusionRetriever(
29
  [vector_retriever, bm25_retriever],
30
+ similarity_top_k=70,
31
  num_queries=1
32
  )
33