opc_agent/agent-craft
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore: minor adjustments

Browse Source
This commit is contained in:
Annyfee
2026-01-16 22:02:28 +08:00
parent cf504582be
commit d2648fc2a1
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
m07_rag_advanced/s03_reranker.py
+1 -1
View File
@@ -38,7 +38,7 @@ print('---Chroma数据库已加载---\n')
# --- 模块 B (R-A-G Flow) --- # --- 模块 B (R-A-G Flow) ---
# 1. R-检索--强化版 # 1. R-检索--强化版
# 1.1 基础检索器(Base Retriever) - '粗召回' # 1.1 基础检索器(Base Retriever) - '粗召回'
base_retriever = db.as_retriever(search_kwargs={"k":50}) # K调大到60 base_retriever = db.as_retriever(search_kwargs={"k":50}) # K调大到50
# 1.2 Reranker (重排器) - "精排序" -- 首次运行需要耗时下载 # 1.2 Reranker (重排器) - "精排序" -- 首次运行需要耗时下载
print('正在加载 Reranker模型 (bge-reranker-base)...') print('正在加载 Reranker模型 (bge-reranker-base)...')
encoder = HuggingFaceCrossEncoder(model_name="BAAI/bge-reranker-base") # 加载Ranker模型 encoder = HuggingFaceCrossEncoder(model_name="BAAI/bge-reranker-base") # 加载Ranker模型