update

HowieHwong · Apr 21, 2024 · c6465d3 · c6465d3
1 parent e680cf7
commit c6465d3
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):