From a30458e780d01179a55472afe9e00e39e23322b3 Mon Sep 17 00:00:00 2001
From: JINO-ROHIT <find.jinorohit@gmail.com>
Date: Thu, 28 Nov 2024 21:26:56 +0530
Subject: [PATCH] revert float type cast manually in
 BinaryClassificationEvaluator

---
 .../evaluation/BinaryClassificationEvaluator.py  | 16 ++++++++--------
 1 file changed, 8 insertions(+), 8 deletions(-)

diff --git a/sentence_transformers/evaluation/BinaryClassificationEvaluator.py b/sentence_transformers/evaluation/BinaryClassificationEvaluator.py
index 4a701279d..e0e00ebc9 100644
--- a/sentence_transformers/evaluation/BinaryClassificationEvaluator.py
+++ b/sentence_transformers/evaluation/BinaryClassificationEvaluator.py
@@ -288,14 +288,14 @@ def compute_metrices(self, model: SentenceTransformer) -> dict[str, dict[str, fl
             logger.info(f"Matthews Correlation with {name}: {mcc * 100:.2f}\n")
 
             output_scores[similarity_fn_name] = {
-                "accuracy": float(acc),
-                "accuracy_threshold": float(acc_threshold),
-                "f1": float(f1),
-                "f1_threshold": float(f1_threshold),
-                "precision": float(precision),
-                "recall": float(recall),
-                "ap": float(ap),
-                "mcc": float(mcc),
+                "accuracy": acc,
+                "accuracy_threshold": acc_threshold,
+                "f1": f1,
+                "f1_threshold": f1_threshold,
+                "precision": precision,
+                "recall": recall,
+                "ap": ap,
+                "mcc": mcc,
             }
 
         return output_scores