Spaces:
Sleeping
Sleeping
update hybrid_retriever_tool file
Browse files
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()) / (
|
| 51 |
-
sem_norm = (sem_scores - sem_scores.min()) / (
|
| 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
|