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

update genetic algorithm

Browse Source
This commit is contained in:
Jin Yong Yoo
2020-06-29 05:14:12 -04:00
parent 21133e0296
commit d4c7030328
6 changed files with 73 additions and 55 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
textattack/shared/attack.py
View File

@@ -76,9 +76,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
if isinstance(self.search_method, textattack.search_methods.GeneticAlgorithm):
self.search_method.filter_transformations = self.filter_transformations
self.search_method.filter_transformations = self.filter_transformations
def get_transformations(self, current_text, original_text=None, **kwargs):
"""