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

add BAE and BERT-Attack recipes

Browse Source
This commit is contained in:
Jack Morris
2020-06-28 16:00:12 -04:00
parent 3f15bb34f8
commit 7daadbc899
9 changed files with 228 additions and 300 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
textattack/commands/attack/attack_args.py
View File

@@ -2,6 +2,8 @@ import textattack
ATTACK_RECIPE_NAMES = {
"alzantot": "textattack.attack_recipes.Alzantot2018",
"bae": "textattack.attack_recipes.BAEGarg2019",
"bert-attack": "textattack.attack_recipes.BERTAttackLi2020",
"deepwordbug": "textattack.attack_recipes.DeepWordBugGao2018",
"hotflip": "textattack.attack_recipes.HotFlipEbrahimi2017",
"kuleshov": "textattack.attack_recipes.Kuleshov2017",