MikeMann commited on
Commit
402adb3
·
1 Parent(s): 574cec8

adjusted K-params

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -345,7 +345,7 @@ class BSIChatbot:
345
  if bm25_retriever is None:
346
  bm25_retriever = BM25Retriever.from_documents(allDocs)
347
  #TODO!
348
- retriever_k=15
349
  bm25_retriever.k= retriever_k
350
  vectordb = vectorstore.as_retriever(search_kwargs={"k":retriever_k})
351
  ensemble_retriever = EnsembleRetriever(retrievers=[bm25_retriever, vectordb], weights=[0.5, 0.5])
@@ -380,7 +380,7 @@ class BSIChatbot:
380
  #new print(retrieved_chunks_text)
381
  with torch.no_grad():
382
  print("reranking chunks..")
383
- retrieved_chunks_text = rerankingModel.rerank(query, retrieved_chunks_text, k=15)
384
  #newprint("DBG:retrieved_chunks_text after rerank")
385
  #newprint(type(retrieved_chunks_text))
386
  #newprint(retrieved_chunks_text)
 
345
  if bm25_retriever is None:
346
  bm25_retriever = BM25Retriever.from_documents(allDocs)
347
  #TODO!
348
+ retriever_k=25
349
  bm25_retriever.k= retriever_k
350
  vectordb = vectorstore.as_retriever(search_kwargs={"k":retriever_k})
351
  ensemble_retriever = EnsembleRetriever(retrievers=[bm25_retriever, vectordb], weights=[0.5, 0.5])
 
380
  #new print(retrieved_chunks_text)
381
  with torch.no_grad():
382
  print("reranking chunks..")
383
+ retrieved_chunks_text = rerankingModel.rerank(query, retrieved_chunks_text, k=20)
384
  #newprint("DBG:retrieved_chunks_text after rerank")
385
  #newprint(type(retrieved_chunks_text))
386
  #newprint(retrieved_chunks_text)