调参

tangwang
1 parent f8219b5e
Showing 2 changed files with 8 additions and 8 deletions Show diff stats
config/config.yaml
search/es_query_builder.py
@@ -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
@@ -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,
                     }
                 }
             )