sbmaruf commited on
Commit
1713ff5
·
1 Parent(s): c2b7f10

update avg on dashboard

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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
- overall_score = sum(scores) / len(scores)
 
 
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