fisherman611 commited on
Commit
f193e4b
·
verified ·
1 Parent(s): 5695300

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +7 -7
config.py CHANGED
@@ -30,21 +30,21 @@ class Config:
30
 
31
  # RAG Configuration
32
  TOP_K_RETRIEVAL = 15
33
- BM25_TOP_K = 15
34
- SIMILARITY_THRESHOLD = 0.5
35
 
36
  # Reranker Configuration
37
  ENABLE_RERANKING = True
38
  RERANKER_MODEL = "cross-encoder/ms-marco-MiniLM-L-6-v2"
39
- RERANKER_TOP_K = 10 # Final number of documents after reranking
40
- RERANK_BEFORE_RETRIEVAL_TOP_K = 15 # Retrieve more docs before ranking
41
- USE_SCORE_FUSION = True # Use fusion between original and reranker scores
42
- RERANKER_FUSION_ALPHA = 0.6 # Weight for reranker score in fusion (0-1)
43
 
44
  # Google Search Configuration
45
  ENABLE_GOOGLE_SEARCH = True
46
  GOOGLE_SEARCH_RESULTS_COUNT = 10
47
- MIN_SIMILARITY_FOR_LEGAL_DOCS = 0.3
48
 
49
  # Question Refinement Configuration
50
  ENABLE_QUESTION_REFINEMENT = True
 
30
 
31
  # RAG Configuration
32
  TOP_K_RETRIEVAL = 15
33
+ BM25_TOP_K = 20
34
+ SIMILARITY_THRESHOLD = 0.25
35
 
36
  # Reranker Configuration
37
  ENABLE_RERANKING = True
38
  RERANKER_MODEL = "cross-encoder/ms-marco-MiniLM-L-6-v2"
39
+ RERANKER_TOP_K = 10
40
+ RERANK_BEFORE_RETRIEVAL_TOP_K = 25
41
+ USE_SCORE_FUSION = True
42
+ RERANKER_FUSION_ALPHA = 0.8
43
 
44
  # Google Search Configuration
45
  ENABLE_GOOGLE_SEARCH = True
46
  GOOGLE_SEARCH_RESULTS_COUNT = 10
47
+ MIN_SIMILARITY_FOR_LEGAL_DOCS = 0.15
48
 
49
  # Question Refinement Configuration
50
  ENABLE_QUESTION_REFINEMENT = True