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