alihan/textattack-nlp-transformer
1
0
Fork 0
mirror of https://github.com/QData/TextAttack.git synced 2021-10-13 00:05:06 +03:00
Code Issues Projects Releases Wiki Activity

merge with master

Browse Source
This commit is contained in:
uvafan
2020-07-01 00:56:33 -04:00
parent a2532198a5 c6c40a15a8
commit 7e9f3ece13
45 changed files with 502 additions and 351 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
textattack/shared/attack.py
View File

@@ -78,6 +78,7 @@ class Attack:
# Give search method access to functions for getting transformations and evaluating them
self.search_method.get_transformations = self.get_transformations
self.search_method.get_goal_results = self.goal_function.get_results
self.search_method.filter_transformations = self.filter_transformations
def get_transformations(self, current_text, original_text=None, **kwargs):
"""
@@ -105,7 +106,7 @@ class Attack:
**kwargs,
)
)
return self._filter_transformations(
return self.filter_transformations(
transformed_texts, current_text, original_text
)
@@ -135,7 +136,7 @@ class Attack:
self.constraints_cache[(current_text, filtered_text)] = True
return filtered_texts
def _filter_transformations(
def filter_transformations(
self, transformed_texts, current_text, original_text=None
):
"""