File size: 830 Bytes
4943752 |
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 |
"""
Goal Function for TextToText
-------------------------------------------------------
"""
from textattack.goal_function_results import TextToTextGoalFunctionResult
from textattack.goal_functions import GoalFunction
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
|