Fixed typo

Fixed a type in `cos_sim_2d(x, y)`
jonylentzmc · Mar 20, 2023 · 78f970f · 78f970f
1 parent cd0f1c9
commit 78f970f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/retrieval/wikipedia.py b/retrieval/wikipedia.py
@@ -20,7 +20,7 @@ def mean_pooling(token_embeddings, mask):
 
 def cos_sim_2d(x, y):
     norm_x = x / np.linalg.norm(x, axis=1, keepdims=True)
-    norm_y = y / np.linalg.norm(x, axis=1, keepdims=True)
+    norm_y = y / np.linalg.norm(y, axis=1, keepdims=True)
     return np.matmul(norm_x, norm_y.T)