epsilon-compositionality

commit 5eff8e8f5cc73ca14364ae980b875f0c43d43d51
parent 57a8735d2128141ede3e7b5ccb176e37aa8ec864
Author: Sondre Wold <sondrewold1@gmail.com>
Date:   Thu, 28 Aug 2025 12:17:54 +0200

get float from sent-transformers

Diffstat:
M
experiment.py
 | 
8
++++----

1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/experiment.py b/experiment.py
@@ -200,7 +200,7 @@ class S_BERT(Model):
         return self.encoder.encode([string])
 
     def calculate_similarity(self, a: torch.Tensor, b: torch.Tensor) -> float:
-        return self.encoder.similarity(a, b)
+        return self.encoder.similarity(a, b).item()
 
 
 @ModelFactory("S-GPT")
@@ -215,7 +215,7 @@ class S_GPT(Model):
         return self.encoder.encode([string])
 
     def calculate_similarity(self, a: torch.Tensor, b: torch.Tensor) -> float:
-        return self.encoder.similarity(a, b)
+        return self.encoder.similarity(a, b).item()
 
 
 @ModelFactory("E5")
@@ -235,7 +235,7 @@ class E5(Model):
         return self.encoder.encode([prepared_string])
 
     def calculate_similarity(self, a: torch.Tensor, b: torch.Tensor) -> float:
-        return self.encoder.similarity(a, b)
+        return self.encoder.similarity(a, b).item()
 
 
 @ModelFactory("QWEN")
@@ -253,7 +253,7 @@ class QWEN(Model):
         return self.encoder.encode([prepared_string])
 
     def calculate_similarity(self, a: torch.Tensor, b: torch.Tensor) -> float:
-        return self.encoder.similarity(a, b)
+        return self.encoder.similarity(a, b).item()
 
 
 def find_diff(a: str, b: str) -> list[str] | None: