Spaces:
Running
Running
update avg on dashboard
Browse files
app.py
CHANGED
@@ -904,10 +904,11 @@ def main():
|
|
904 |
# Calculate average safety score across all selected models
|
905 |
overall_scores = {}
|
906 |
for model in selected_models:
|
907 |
-
# Calculate an overall score (average of all category safe scores)
|
908 |
categories = list(category_data["categories"].keys())
|
909 |
scores = [category_data["categories"][category][model]["safe"]*100 for category in categories]
|
910 |
-
|
|
|
|
|
911 |
overall_scores[model] = overall_score
|
912 |
|
913 |
# Calculate the average across all selected models
|
|
|
904 |
# Calculate average safety score across all selected models
|
905 |
overall_scores = {}
|
906 |
for model in selected_models:
|
|
|
907 |
categories = list(category_data["categories"].keys())
|
908 |
scores = [category_data["categories"][category][model]["safe"]*100 for category in categories]
|
909 |
+
category_weights = [category_data["categories"][category]["total"] for category in categories]
|
910 |
+
total_examples = sum(category_weights)
|
911 |
+
overall_score = sum(scores[i] * category_weights[i] / total_examples for i in range(len(scores)))
|
912 |
overall_scores[model] = overall_score
|
913 |
|
914 |
# Calculate the average across all selected models
|