cicboy commited on
Commit
c659851
·
1 Parent(s): 1596fb6

update hybrid_retriever_tool file

Browse files
Files changed (1) hide show
  1. tools/hybrid_retriever_tool.py +2 -2
tools/hybrid_retriever_tool.py CHANGED
@@ -47,8 +47,8 @@ class HybridRetrieverTool(RagTool):
47
  sem_scores = np.dot(emb_corpus, emb_query)
48
 
49
  # Normalize scores
50
- bm25_norm = (bm25_scores - bm25_scores.min()) / (bm25_scores.ptp() + 1e-8)
51
- sem_norm = (sem_scores - sem_scores.min()) / (sem_scores.ptp() + 1e-8)
52
 
53
  # Weighted fusion
54
  hybrid_scores = self.alpha * sem_norm + (1 - self.alpha) * bm25_norm
 
47
  sem_scores = np.dot(emb_corpus, emb_query)
48
 
49
  # Normalize scores
50
+ bm25_norm = (bm25_scores - bm25_scores.min()) / (np.ptp(bm25_scores) + 1e-8)
51
+ sem_norm = (sem_scores - sem_scores.min()) / (np.ptp(sem_scores) + 1e-8)
52
 
53
  # Weighted fusion
54
  hybrid_scores = self.alpha * sem_norm + (1 - self.alpha) * bm25_norm