github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-04 02:25:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(rag): Fix CrossEncoderRanker bug of EmbeddingRetriever (#1504)

Browse Source 
Co-authored-by: aries_ckt <916701291@qq.com>
This commit is contained in:
ivanzhu
2024-05-10 13:32:58 +08:00
committed by GitHub
parent 87a67cce7d
commit 8eb64d7311
3 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
dbgpt/rag/retriever/rerank.py
View File

@@ -219,7 +219,7 @@ class CrossEncoderRanker(Ranker):
rank_scores = self._model.predict(sentences=query_content_pairs)
for candidate, score in zip(candidates_with_scores, rank_scores):
candidate.score = score
candidate.score = float(score)
new_candidates_with_scores = sorted(
candidates_with_scores, key=lambda x: x.score, reverse=True