From 418b6a4a5e62dce57b02b9c13b7b66b8ba6d0dbc Mon Sep 17 00:00:00 2001 From: tangwang Date: Mon, 30 Mar 2026 22:33:10 +0800 Subject: [PATCH] 调参 --- config/config.yaml | 8 ++++---- search/es_query_builder.py | 8 ++++---- 2 files changed, 8 insertions(+), 8 deletions(-) diff --git a/config/config.yaml b/config/config.yaml index 43aeb0a..6093e4b 100644 --- a/config/config.yaml +++ b/config/config.yaml @@ -248,8 +248,8 @@ coarse_rank: # 精排配置(轻量 reranker) fine_rank: - enabled: true - input_window: 240 + enabled: false + input_window: 160 output_window: 80 timeout_sec: 10.0 rerank_query_template: "{query}" @@ -259,7 +259,7 @@ fine_rank: # 重排配置(provider/URL 在 services.rerank) rerank: enabled: true - rerank_window: 80 + rerank_window: 160 timeout_sec: 15.0 weight_es: 0.4 weight_ai: 0.6 @@ -457,7 +457,7 @@ services: model_name: "BAAI/bge-reranker-v2-m3" device: null use_fp16: true - batch_size: 64 + batch_size: 80 max_length: 160 cache_dir: "./model_cache" enable_warmup: true diff --git a/search/es_query_builder.py b/search/es_query_builder.py index 0292d9f..4c20e93 100644 --- a/search/es_query_builder.py +++ b/search/es_query_builder.py @@ -38,9 +38,9 @@ class ESQueryBuilder: knn_text_num_candidates_long: int = 500, knn_image_k: int = 120, knn_image_num_candidates: int = 400, - base_minimum_should_match: str = "70%", - translation_minimum_should_match: str = "70%", - keywords_minimum_should_match: str = "50%", + base_minimum_should_match: str = "66%", + translation_minimum_should_match: str = "66%", + keywords_minimum_should_match: str = "60%", translation_boost: float = 0.4, tie_breaker_base_query: float = 0.9, best_fields_boosts: Optional[Dict[str, float]] = None, @@ -549,7 +549,7 @@ class ESQueryBuilder: "query": kw, "fields": combined_fields, "minimum_should_match": self.keywords_minimum_should_match, - "boost": 0.6, + "boost": 0.8, } } ) -- libgit2 0.21.2