minor bug fix

leaderboard.py

@@ -25,7 +25,7 @@ def map_df(full_df):
   pred_mapping = {'fake':1, 'real': 0}
 
 
-  full_df['label'] = full_df['ground_truth'].map(gnd_truth_mapping)
+  full_df['gnd_truth'] = full_df['label'].map(gnd_truth_mapping)
   full_df['pred'] = full_df['type'].map(pred_mapping)
 
   return full_df
@@ -39,19 +39,19 @@ def get_duration_scores(df):
   for c in columns:
     
     if c == 'Overall':
-      mask = df.label == 0
+      mask = df.gnd_truth == 0
     elif c == 'Under 26 s':
-      mask = (df.label == 0) & (df.duration < 26)
+      mask = (df.gnd_truth == 0) & (df.duration < 26)
     elif c == '55 s':
-      mask = (df.label == 0) & (df.duration >= 26) & (df.duration < 56)
+      mask = (df.gnd_truth == 0) & (df.duration >= 26) & (df.duration < 56)
     elif c == '125 s':
-      mask = (df.label == 0) & (df.duration >= 56) & (df.duration < 126)
+      mask = (df.gnd_truth == 0) & (df.duration >= 56) & (df.duration < 126)
     else:
       raise ValueError
     sel_df = df[mask]
 
     samples_tested.append(len(sel_df))
-    acc_scores.append(round(accuracy_score(sel_df.label.values, sel_df.pred.values), 3))
+    acc_scores.append(round(accuracy_score(sel_df.gnd_truth.values, sel_df.pred.values), 3))
   
   lb = pd.DataFrame({"Sample": columns, "Num Samples": samples_tested, "Accuracy": acc_scores})
   return lb