Spaces:
Sleeping
Sleeping
File size: 1,829 Bytes
4a1df2e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 |
"""
Attack Logs to CSV
========================
"""
import csv
import pandas as pd
from textattack.shared import AttackedText, logger
from .logger import Logger
class CSVLogger(Logger):
"""Logs attack results to a CSV."""
def __init__(self, filename="results.csv", color_method="file"):
logger.info(f"Logging to CSV at path {filename}")
self.filename = filename
self.color_method = color_method
self.row_list = []
self._flushed = True
def log_attack_result(self, result):
original_text, perturbed_text = result.diff_color(self.color_method)
original_text = original_text.replace("\n", AttackedText.SPLIT_TOKEN)
perturbed_text = perturbed_text.replace("\n", AttackedText.SPLIT_TOKEN)
result_type = result.__class__.__name__.replace("AttackResult", "")
row = {
"original_text": original_text,
"perturbed_text": perturbed_text,
"original_score": result.original_result.score,
"perturbed_score": result.perturbed_result.score,
"original_output": result.original_result.output,
"perturbed_output": result.perturbed_result.output,
"ground_truth_output": result.original_result.ground_truth_output,
"num_queries": result.num_queries,
"result_type": result_type,
}
self.row_list.append(row)
self._flushed = False
def flush(self):
self.df = pd.DataFrame.from_records(self.row_list)
self.df.to_csv(self.filename, quoting=csv.QUOTE_NONNUMERIC, index=False)
self._flushed = True
def close(self):
# self.fout.close()
super().close()
def __del__(self):
if not self._flushed:
logger.warning("CSVLogger exiting without calling flush().")
|