Spaces:
Sleeping
Sleeping
Commit ·
c63d524
1
Parent(s): 37d88ed
added 35 semantic similarity
Browse files- 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=
|
| 25 |
-
similarity_cutoff=0.
|
| 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.
|
| 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 |
|