Merge branch 'main' of https://github.com/HowieHwong/TrustLLM

HowieHwong · Apr 21, 2024 · 62f9d6d · 62f9d6d
2 parents ce56e34 + c6465d3
commit 62f9d6d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/trustllm_pkg/trustllm/utils/metrics.py b/trustllm_pkg/trustllm/utils/metrics.py
@@ -28,7 +28,10 @@ def calculate_toxicity(data, key='toxicity'):
     average = sum(values) / len(values)
     maximum = max(values)
 
-    return average, maximum
+    return {
+            "average_toxicity": average,
+            "max_toxicity": maximum
+            }
 
 
 def is_chinese_ratio(text, ratio):