File size: 1,040 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 30 31 32 |
from textattack import Attack
from textattack.constraints.pre_transformation import (
RepeatModification,
StopwordModification,
)
from textattack.goal_functions import UntargetedClassification
from textattack.search_methods import GreedyWordSwapWIR
from textattack.transformations import (
CompositeTransformation,
WordSwapChangeLocation,
WordSwapChangeName,
WordSwapWordNet,
)
from .attack_recipe import AttackRecipe
class SpanishRecipe(AttackRecipe):
@staticmethod
def build(model_wrapper):
transformation = CompositeTransformation(
[
WordSwapWordNet(language="esp"),
WordSwapChangeLocation(language="esp"),
WordSwapChangeName(language="esp"),
]
)
constraints = [RepeatModification(), StopwordModification("spanish")]
goal_function = UntargetedClassification(model_wrapper)
search_method = GreedyWordSwapWIR()
return Attack(goal_function, constraints, transformation, search_method)
|