fix: conversion of cosine distance to cosine similarity is incorrect

piskvorky · Feb 20, 2023 · 4a33647 · 4a33647
1 parent f260d1e
commit 4a33647
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))]