Fix: Minor bug in mean calculation.

Anindyadeep · Nov 14, 2023 · 0800a6c · 0800a6c
1 parent 03edce8
commit 0800a6c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepeval/metrics/scoring.py b/deepeval/metrics/scoring.py
@@ -255,7 +255,7 @@ def neural_toxic_score(
         else:
             detoxify_model = Detoxify("original", device=device)
         toxicity_score_dict = detoxify_model.predict(prediction)
-        mean_toxicity_score = list(toxicity_score_dict.values()) / len(
+        mean_toxicity_score = sum(list(toxicity_score_dict.values())) / len(
             toxicity_score_dict
         )
         return mean_toxicity_score, toxicity_score_dict