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 branch 'master' of github.com:QData/TextAttack into faster-alz-lm

Browse Source
This commit is contained in:
Jack Morris
2020-06-29 13:52:41 -04:00
parent 05139cf828 03a9585022
commit db5307432f
20 changed files with 158 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -9,6 +9,7 @@ ATTACK_RECIPE_NAMES = {
"seq2sick": "textattack.attack_recipes.Seq2SickCheng2018BlackBox",
"textbugger": "textattack.attack_recipes.TextBuggerLi2018",
"textfooler": "textattack.attack_recipes.TextFoolerJin2019",
"pwws": "textattack.attack_recipes.PWWSRen2019",
}
#
@@ -55,9 +56,13 @@ HUGGINGFACE_DATASET_BY_MODEL = {
("glue", "wnli", "validation"),
),
"bert-base-uncased-mr": (
"textattack/bert-base-uncased-rotten_tomatoes",
"textattack/bert-base-uncased-rotten-tomatoes",
("rotten_tomatoes", None, "test"),
),
"bert-base-uncased-snli": (
"textattack/bert-base-uncased-snli",
("snli", None, "test", [1, 2, 0]),
),
#
# distilbert-base-cased
#
@@ -145,14 +150,21 @@ HUGGINGFACE_DATASET_BY_MODEL = {
("glue", "wnli", "validation"),
),
"roberta-base-mr": (
"textattack/roberta-base-rotten_tomatoes",
"textattack/roberta-base-rotten-tomatoes",
("rotten_tomatoes", None, "test"),
),
#
# albert-base-v2 (ALBERT is cased by default)
#
"albert-base-v2-mr": (
"textattack/albert-base-v2-rotten_tomatoes",
"textattack/albert-base-v2-rotten-tomatoes",
("rotten_tomatoes", None, "test"),
),
#
# xlnet-base-cased
#
"xlnet-base-cased-mr": (
"textattack/xlnet-base-cased-rotten-tomatoes",
("rotten_tomatoes", None, "test"),
),
}