MrSimple07 commited on
Commit
d6920c6
·
1 Parent(s): ee54fb7

topk query = 70 + 0.55 sim cut off + table chunk size= 4000

Browse files
Files changed (1) hide show
  1. index_retriever.py +3 -3
index_retriever.py CHANGED
@@ -12,7 +12,7 @@ def create_vector_index(documents):
12
  log_message("Строю векторный индекс")
13
  return VectorStoreIndex.from_documents(documents)
14
 
15
- def rerank_nodes(query, nodes, reranker, top_k=20, min_score_threshold=0.5):
16
  if not nodes or not reranker:
17
  return nodes[:top_k]
18
 
@@ -46,12 +46,12 @@ def create_query_engine(vector_index):
46
 
47
  bm25_retriever = BM25Retriever.from_defaults(
48
  docstore=vector_index.docstore,
49
- similarity_top_k=50
50
  )
51
 
52
  vector_retriever = VectorIndexRetriever(
53
  index=vector_index,
54
- similarity_top_k=50,
55
  similarity_cutoff=0.55
56
  )
57
 
 
12
  log_message("Строю векторный индекс")
13
  return VectorStoreIndex.from_documents(documents)
14
 
15
+ def rerank_nodes(query, nodes, reranker, top_k=25, min_score_threshold=0.5):
16
  if not nodes or not reranker:
17
  return nodes[:top_k]
18
 
 
46
 
47
  bm25_retriever = BM25Retriever.from_defaults(
48
  docstore=vector_index.docstore,
49
+ similarity_top_k=70
50
  )
51
 
52
  vector_retriever = VectorIndexRetriever(
53
  index=vector_index,
54
+ similarity_top_k=70,
55
  similarity_cutoff=0.55
56
  )
57