MrSimple07 commited on
Commit
c63d524
·
1 Parent(s): 37d88ed

added 35 semantic similarity

Browse files
Files changed (1) hide show
  1. index_retriever.py +3 -3
index_retriever.py CHANGED
@@ -21,8 +21,8 @@ def create_query_engine(vector_index):
21
 
22
  vector_retriever = VectorIndexRetriever(
23
  index=vector_index,
24
- similarity_top_k=35, # Increased
25
- similarity_cutoff=0.65 # Slightly lower for recall
26
  )
27
 
28
  hybrid_retriever = QueryFusionRetriever(
@@ -74,7 +74,7 @@ def rerank_nodes(query, nodes, reranker, top_k=20, min_score_threshold=0.5, dive
74
  log_message("Нет узлов после фильтрации, снижаю порог")
75
  scored_nodes = list(zip(nodes, scores))
76
  scored_nodes.sort(key=lambda x: x[1], reverse=True)
77
- min_score_threshold = scored_nodes[0][1] * 0.5 # 50% of top score
78
  scored_nodes = [(node, score) for node, score in scored_nodes
79
  if score >= min_score_threshold]
80
 
 
21
 
22
  vector_retriever = VectorIndexRetriever(
23
  index=vector_index,
24
+ similarity_top_k=30, # Increased
25
+ similarity_cutoff=0.7 # Slightly lower for recall
26
  )
27
 
28
  hybrid_retriever = QueryFusionRetriever(
 
74
  log_message("Нет узлов после фильтрации, снижаю порог")
75
  scored_nodes = list(zip(nodes, scores))
76
  scored_nodes.sort(key=lambda x: x[1], reverse=True)
77
+ min_score_threshold = scored_nodes[0][1] * 0.6 # 60% of top score
78
  scored_nodes = [(node, score) for node, score in scored_nodes
79
  if score >= min_score_threshold]
80