Spaces:
Sleeping
Sleeping
Commit ·
f49e798
1
Parent(s): 9e1a00e
top k in reranker = 30, bm = 20
Browse files- index_retriever.py +1 -1
- utils.py +1 -1
index_retriever.py
CHANGED
|
@@ -16,7 +16,7 @@ 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(
|
|
|
|
| 16 |
try:
|
| 17 |
bm25_retriever = BM25Retriever.from_defaults(
|
| 18 |
docstore=vector_index.docstore,
|
| 19 |
+
similarity_top_k=20
|
| 20 |
)
|
| 21 |
|
| 22 |
vector_retriever = VectorIndexRetriever(
|
utils.py
CHANGED
|
@@ -243,7 +243,7 @@ def answer_question(question, query_engine, reranker, current_model, chunks_df=N
|
|
| 243 |
question,
|
| 244 |
retrieved_nodes,
|
| 245 |
reranker,
|
| 246 |
-
top_k=
|
| 247 |
min_score_threshold=0.5,
|
| 248 |
diversity_penalty=0.3
|
| 249 |
)
|
|
|
|
| 243 |
question,
|
| 244 |
retrieved_nodes,
|
| 245 |
reranker,
|
| 246 |
+
top_k=30,
|
| 247 |
min_score_threshold=0.5,
|
| 248 |
diversity_penalty=0.3
|
| 249 |
)
|