alihan/textattack-nlp-transformer
1
0
Fork 0
mirror of https://github.com/QData/TextAttack.git synced 2021-10-13 00:05:06 +03:00
Code Issues Projects Releases Wiki Activity

more mturk prep changes

Browse Source
This commit is contained in:
uvafan
2019-12-05 18:59:53 -05:00
parent 4de8c7cfae
commit da3fe32c27
11 changed files with 44 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
textattack/loggers/csv_logger.py
View File

@@ -16,7 +16,7 @@ class CSVLogger(Logger):
return
color_method = None if self.plain else 'file'
s1, s2 = result.diff_color(color_method)
row = {'passage_1': s1, 'passage_2': s2}
row = {'passage_1': s1, 'passage_2': s2, 'score_1': result.orig_score, 'score_2': result.perturbed_score, 'label_1': int(result.original_label), 'label_2': int(result.perturbed_label)}
self.df = self.df.append(row, ignore_index=True)
def flush(self):