Source code for textattack.goal_functions.text.text_to_text_goal_function

"""

Goal Function for TextToText
-------------------------------------------------------
"""


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.""" return outputs.flatten() def _get_displayed_output(self, raw_output): return raw_output