feat: 重排处理
This commit is contained in:
parent
b2fcd4a69c
commit
5f3fc6ff07
|
@ -50,11 +50,7 @@ def main():
|
||||||
}
|
}
|
||||||
|
|
||||||
# 重排配置
|
# 重排配置
|
||||||
rerank_config = {
|
rerank_config = {"enabled": True, "method": "cross_encoder", "top_k": 3}
|
||||||
"enabled": True,
|
|
||||||
"method": "similarity",
|
|
||||||
"top_k": 3
|
|
||||||
}
|
|
||||||
|
|
||||||
# 初始化RAG系统
|
# 初始化RAG系统
|
||||||
rag = RerankRAG(
|
rag = RerankRAG(
|
||||||
|
|
Loading…
Reference in New Issue