removing redundant label columns from predictions

phosseini · Apr 28, 2022 · 4953eaf · 4953eaf
1 parent c60d946
commit 4953eaf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fine_tuning_copa.py b/fine_tuning_copa.py
@@ -205,6 +205,7 @@ def get_model():
 df_test_dataset = test_dataset.to_pandas()
 df_results['text'] = df_test_dataset['startphrase'] + " [0] " + df_test_dataset['ending0'] + " [1] " + df_test_dataset[
     'ending1']
+df_results['label'] = df_test_dataset['label']
 
 for run in params['random_seeds']:
     setattr(training_args, 'seed', run)
@@ -227,7 +228,6 @@ def get_model():
     predicted = le.inverse_transform(predictions.predictions.argmax(-1))
     labels = le.inverse_transform(test_dataset['label'])
     df_results['predicted_{}'.format(run)] = predicted
-    df_results['labels_{}'.format(run)] = labels
 
     assert accuracy_score(labels, predicted) == accuracy['accuracy']