Zekun Wu commited on
Commit
a6a3f18
1 Parent(s): 9edca9e
Files changed (1) hide show
  1. util/injection.py +1 -0
util/injection.py CHANGED
@@ -91,6 +91,7 @@ def process_scores_multiple(df, num_run, parameters, privilege_label, protect_la
91
  print(f"Processing {len(df)} entries with {num_run} runs each.")
92
  """ Process entries and compute scores concurrently, with progress updates. """
93
  scores = {key: [[] for _ in range(len(df))] for key in ['Privilege', 'Protect', 'Neutral']}
 
94
 
95
  for run in tqdm(range(num_run), desc="Processing runs", unit="run"):
96
  for index, row in tqdm(df.iterrows(), total=len(df), desc="Processing entries", unit="entry"):
 
91
  print(f"Processing {len(df)} entries with {num_run} runs each.")
92
  """ Process entries and compute scores concurrently, with progress updates. """
93
  scores = {key: [[] for _ in range(len(df))] for key in ['Privilege', 'Protect', 'Neutral']}
94
+ print(f"Scores: {scores}")
95
 
96
  for run in tqdm(range(num_run), desc="Processing runs", unit="run"):
97
  for index, row in tqdm(df.iterrows(), total=len(df), desc="Processing entries", unit="entry"):