MrSimple07 commited on
Commit
8befcd1
·
1 Parent(s): e91b5fc

bm = 40, semantic = 50. hybrid = 40

Browse files
Files changed (2) hide show
  1. index_retriever.py +2 -2
  2. utils.py +1 -1
index_retriever.py CHANGED
@@ -16,12 +16,12 @@ def create_query_engine(vector_index):
16
  try:
17
  bm25_retriever = BM25Retriever.from_defaults(
18
  docstore=vector_index.docstore,
19
- similarity_top_k=30
20
  )
21
 
22
  vector_retriever = VectorIndexRetriever(
23
  index=vector_index,
24
- similarity_top_k=45,
25
  similarity_cutoff=0.7
26
  )
27
 
 
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=50,
25
  similarity_cutoff=0.7
26
  )
27
 
utils.py CHANGED
@@ -279,7 +279,7 @@ def answer_question(question, query_engine, reranker, current_model, chunks_df=N
279
  question,
280
  all_nodes,
281
  reranker,
282
- top_k=25,
283
  min_score_threshold=0.5,
284
  diversity_penalty=0.3
285
  )
 
279
  question,
280
  all_nodes,
281
  reranker,
282
+ top_k=20,
283
  min_score_threshold=0.5,
284
  diversity_penalty=0.3
285
  )