Spaces:
Sleeping
Sleeping
Commit
·
d6920c6
1
Parent(s):
ee54fb7
topk query = 70 + 0.55 sim cut off + table chunk size= 4000
Browse files- 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=
|
| 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 |
)
|
| 51 |
|
| 52 |
vector_retriever = VectorIndexRetriever(
|
| 53 |
index=vector_index,
|
| 54 |
-
similarity_top_k=
|
| 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 |
|