fix: conversion of cosine distance to cosine similarity is incorrect (#…

…3441) Co-authored-by: Mona Shaheen <mona.shaheen@dynamicyield.com>
piskvorky · Mar 3, 2023 · 63fce18 · 63fce18
1 parent f260d1e
commit 63fce18
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gensim/similarities/annoy.py b/gensim/similarities/annoy.py
@@ -185,4 +185,4 @@ def most_similar(self, vector, num_neighbors):
         ids, distances = self.index.get_nns_by_vector(
             vector, num_neighbors, include_distances=True)
 
-        return [(self.labels[ids[i]], 1 - distances[i] / 2) for i in range(len(ids))]
+        return [(self.labels[ids[i]], 1 - distances[i] ** 2 / 2) for i in range(len(ids))]