1
0
mirror of https://github.com/QData/TextAttack.git synced 2021-10-13 00:05:06 +03:00

add outputting to csv

This commit is contained in:
uvafan
2019-12-03 19:10:35 -05:00
parent 23a695b7fa
commit f4e833f21a
5 changed files with 47 additions and 3 deletions

View File

@@ -0,0 +1,20 @@
import sys
import os
import pandas as pd
import csv
from textattack.loggers import Logger
class CSVLogger(Logger):
def __init__(self, filename='results.csv', plain=False):
self.filename = filename
self.plain = plain
self.df = pd.DataFrame()
def log_attack_result(self, result, examples_completed):
color_method = None if self.plain else 'file'
s1, s2 = result.diff_color(color_method)
row = {'passage_1': s1, 'passage_2': s2}
self.df = self.df.append(row, ignore_index=True)
def flush(self):
self.df.to_csv(self.filename, quoting=csv.QUOTE_NONNUMERIC)