update

util/injection.py

@@ -88,7 +88,9 @@ def process_scores_multiple(df, num_run, parameters, privilege_label, protect_la
 
     print(f"Processing {len(df)} entries with {num_run} runs each.")
     """ Process entries and compute scores concurrently, with progress updates. """
-    scores = {key: [[] for _ in range(len(df))] for key in ['Privilege', 'Protect', 'Neutral']}
+    scores = {key: [[] for _ in range(len(df))] for key in ['Privilege_characteristics', 'Privilege_normal',
+                                                             'Protect_characteristics', 'Protect_normal',
+                                                             'Neutral_characteristics', 'Neutral_normal']}
     print(f"Scores: {scores}")
 
     for run in tqdm(range(num_run), desc="Processing runs", unit="run"):