"""
Goal Function for TextToText
-------------------------------------------------------
"""
import numpy as np
from textattack.goal_function_results import TextToTextGoalFunctionResult
from textattack.goal_functions import GoalFunction
[docs]class TextToTextGoalFunction(GoalFunction):
"""A goal function defined on a model that outputs text.
model: The PyTorch or TensorFlow model used for evaluation.
original_output: the original output of the model
"""
def _goal_function_result_type(self):
"""Returns the class of this goal function's results."""
return TextToTextGoalFunctionResult
def _process_model_outputs(self, _, outputs):
"""Processes and validates a list of model outputs."""
if isinstance(outputs, np.ndarray):
return outputs.flatten()
else:
return outputs
def _get_displayed_output(self, raw_output):
return raw_output