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

tests pass

Browse Source
This commit is contained in:
Jack Morris
2020-05-06 23:37:06 -04:00
parent f140541560
commit c446e869d0
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
textattack/constraints/semantics/word_embedding_distance.py
View File

@@ -28,7 +28,7 @@ class WordEmbeddingDistance(Constraint):
self.min_cos_sim = min_cos_sim self.min_cos_sim = min_cos_sim
self.max_mse_dist = max_mse_dist self.max_mse_dist = max_mse_dist
def call_many(self, x, x_adv_list, original_word=None): def call_many(self, x, x_adv_list, original_text=None):
""" Returns each `x_adv` from `x_adv_list` where `C(x,x_adv)` is True. """ Returns each `x_adv` from `x_adv_list` where `C(x,x_adv)` is True.
""" """
return [x_adv for x_adv in x_adv_list if self(x, x_adv)] return [x_adv for x_adv in x_adv_list if self(x, x_adv)]
@@ -65,7 +65,7 @@ class WordEmbeddingDistance(Constraint):
self.word_embedding.mse_dist_mat[a][b] = mse_dist self.word_embedding.mse_dist_mat[a][b] = mse_dist
return mse_dist return mse_dist
def __call__(self, x, x_adv): def __call__(self, x, x_adv, original_text=None):
""" Returns true if (x, x_adv) are closer than `self.min_cos_sim` """ Returns true if (x, x_adv) are closer than `self.min_cos_sim`
and `self.max_mse_dist`. """ and `self.max_mse_dist`. """