connor/CrossEncoderRanker
1
2
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update ce_ranker.py

Browse Source
This commit is contained in:
Connor Shorten
2025-11-27 16:41:39 +00:00
parent 313ab4c8c6
commit 316f16630f
1 changed files with 0 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
ce_ranker.py
View File

@@ -59,10 +59,8 @@ class CERankerAgent(PrecompiledAgent):
*[self._score_document(query, doc) for doc in documents] *[self._score_document(query, doc) for doc in documents]
) )
# Sort by the score (descending). If score is True, it comes first.
scored_results.sort(key=lambda x: x[1], reverse=True) scored_results.sort(key=lambda x: x[1], reverse=True)
# Return a list of documents (not joined as a string), up to k results
return [doc for doc, score in scored_results[:k]] return [doc for doc, score in scored_results[:k]]
def __call__(self, query: str, k: int | None = None) -> list[str]: def __call__(self, query: str, k: int | None = None) -> list[str]: