Spaces:
Sleeping
Sleeping
Commit
·
8befcd1
1
Parent(s):
e91b5fc
bm = 40, semantic = 50. hybrid = 40
Browse files- index_retriever.py +2 -2
- utils.py +1 -1
index_retriever.py
CHANGED
|
@@ -16,12 +16,12 @@ def create_query_engine(vector_index):
|
|
| 16 |
try:
|
| 17 |
bm25_retriever = BM25Retriever.from_defaults(
|
| 18 |
docstore=vector_index.docstore,
|
| 19 |
-
similarity_top_k=
|
| 20 |
)
|
| 21 |
|
| 22 |
vector_retriever = VectorIndexRetriever(
|
| 23 |
index=vector_index,
|
| 24 |
-
similarity_top_k=
|
| 25 |
similarity_cutoff=0.7
|
| 26 |
)
|
| 27 |
|
|
|
|
| 16 |
try:
|
| 17 |
bm25_retriever = BM25Retriever.from_defaults(
|
| 18 |
docstore=vector_index.docstore,
|
| 19 |
+
similarity_top_k=40
|
| 20 |
)
|
| 21 |
|
| 22 |
vector_retriever = VectorIndexRetriever(
|
| 23 |
index=vector_index,
|
| 24 |
+
similarity_top_k=50,
|
| 25 |
similarity_cutoff=0.7
|
| 26 |
)
|
| 27 |
|
utils.py
CHANGED
|
@@ -279,7 +279,7 @@ def answer_question(question, query_engine, reranker, current_model, chunks_df=N
|
|
| 279 |
question,
|
| 280 |
all_nodes,
|
| 281 |
reranker,
|
| 282 |
-
top_k=
|
| 283 |
min_score_threshold=0.5,
|
| 284 |
diversity_penalty=0.3
|
| 285 |
)
|
|
|
|
| 279 |
question,
|
| 280 |
all_nodes,
|
| 281 |
reranker,
|
| 282 |
+
top_k=20,
|
| 283 |
min_score_threshold=0.5,
|
| 284 |
diversity_penalty=0.3
|
| 285 |
)
|