yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2025-12-02 18:06:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: removing warning and using permute to reverse the dimensions of a tensor (#9393)

Browse Source 
* using permute to reverse the dimensions of a tensor
This commit is contained in:
David S. Batista 2025-05-15 15:56:09 +01:00 committed by GitHub
parent af073852d0
commit d8ff508c44
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
haystack/components/rankers/sentence_transformers_diversity.py
View File

@ -365,7 +365,10 @@ class SentenceTransformersDiversityRanker:
if idx in selected:
continue
relevance_score = query_similarities[idx]
diversity_score = max(doc_embeddings[idx] @ doc_embeddings[j].T for j in selected)
diversity_score = max(
doc_embeddings[idx] @ doc_embeddings[j].permute(*torch.arange(doc_embeddings[j].ndim - 1, -1, -1))
for j in selected
)
mmr_score = lambda_threshold * relevance_score - (1 - lambda_threshold) * diversity_score
if mmr_score > best_score:
best_score = mmr_score