Zekun Wu commited on
Commit
95dca90
1 Parent(s): 106ffe4
Files changed (1) hide show
  1. util/injection.py +1 -1
util/injection.py CHANGED
@@ -98,7 +98,7 @@ def process_scores_multiple(df, num_run, parameters, privilege_label, protect_la
98
  print(f"Scores: {scores}")
99
 
100
  # Ensure all scores are lists and calculate average scores
101
- for category in ['Privilege_normal', 'Protect_normal','Neutral_normal']:
102
 
103
 
104
  # Ensure the scores are lists and check before assignment
 
98
  print(f"Scores: {scores}")
99
 
100
  # Ensure all scores are lists and calculate average scores
101
+ for category in ['Privilege', 'Protect','Neutral']:
102
 
103
 
104
  # Ensure the scores are lists and check before assignment