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

fix bugs and refactor

Browse Source
This commit is contained in:
Jin Yong Yoo
2020-07-16 09:12:59 -04:00
parent 8f0c443e66
commit d65b5963ed
4 changed files with 73 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
textattack/search_methods/genetic_algorithm.py
View File

@@ -32,8 +32,8 @@ class GeneticAlgorithm(PopulationBasedMethod):
def __init__(
self,
pop_size=20,
max_iters=50,
pop_size=60,
max_iters=20,
temp=0.3,
give_up_if_no_improvement=False,
post_crossover_check=True,
@@ -133,6 +133,10 @@ class GeneticAlgorithm(PopulationBasedMethod):
new_text = x1_text.replace_words_at_indices(
indices_to_replace, words_to_replace
)
indices_to_replace = set(indices_to_replace)
new_text.attack_attrs["modified_indices"] = (
x1_text.attack_attrs["modified_indices"] - indices_to_replace
) | (x2_text.attack_attrs["modified_indices"] & indices_to_replace)
if not self.post_crossover_check or (
new_text.text == x1_text.text or new_text.text == x2_text.text