[WIP] Update CBC example

2021-06-02 00:45:33 +02:00
parent 88cfd5762e
commit d46fe4a393
2 changed files with 7 additions and 10 deletions
--- a/prototorch/models/cbc.py
+++ b/prototorch/models/cbc.py
@@ -16,9 +16,9 @@ def shift_activation(x):
    return (x + 1.0) / 2.0


-def euclidean_similarity(x, y, beta=3):
+def euclidean_similarity(x, y, variance=1.0):
    d = euclidean_distance(x, y)
-    return torch.exp(-d * beta)
+    return torch.exp(-(d * d) / (2 * variance))


 class CosineSimilarity(torch.nn.Module):