1
0
mirror of https://github.com/QData/TextAttack.git synced 2021-10-13 00:05:06 +03:00

entailment works; add peek-dataset command

This commit is contained in:
Jack Morris
2020-06-23 20:22:03 -04:00
parent 69e114099b
commit 5991aa6d4e
15 changed files with 159 additions and 54 deletions

View File

@@ -333,6 +333,8 @@ def parse_dataset_from_args(args):
dataset_args = args.dataset_from_nlp
if ":" in dataset_args:
dataset_args = dataset_args.split(":")
else:
dataset_args = (dataset_args,)
dataset = textattack.datasets.HuggingFaceNLPDataset(
*dataset_args, shuffle=args.shuffle
)