textattack.commands package

TextAttack commands Package

AttackCommand class

class textattack.commands.attack_command.AttackCommand

Bases: TextAttackCommand

The TextAttack attack module:

A command line parser to run an attack from user specifications.

static register_subcommand(main_parser: ArgumentParser)

run(args)

AttackResumeCommand class

class textattack.commands.attack_resume_command.AttackResumeCommand

Bases: TextAttackCommand

The TextAttack attack resume recipe module:

A command line parser to resume a checkpointed attack from user specifications.

static register_subcommand(main_parser: ArgumentParser)

run(args)

AugmentCommand class

class textattack.commands.augment_command.AugmentCommand

Bases: TextAttackCommand

The TextAttack attack module:

A command line parser to run data augmentation from user specifications.

static register_subcommand(main_parser: ArgumentParser)

run(args)

Reads in a CSV, performs augmentation, and outputs an augmented CSV.

Preserves all columns except for the input (augmneted) column.

BenchmarkRecipeCommand class

class textattack.commands.benchmark_recipe_command.BenchmarkRecipeCommand

Bases: TextAttackCommand

The TextAttack benchmark recipe module:

A command line parser to benchmark a recipe from user specifications.

static register_subcommand(main_parser: ArgumentParser)

run(args)

EvalModelCommand class

class textattack.commands.eval_model_command.EvalModelCommand

Bases: TextAttackCommand

The TextAttack model benchmarking module:

A command line parser to evaluatate a model from user specifications.

get_preds(model, inputs)

static register_subcommand(main_parser: ArgumentParser)

run(args)

test_model_on_dataset(args)

class textattack.commands.eval_model_command.ModelEvalArgs(dataset_by_model: str = None, dataset_from_huggingface: str = None, dataset_from_file: str = None, dataset_split: str = None, filter_by_labels: list = None, model: str = None, model_from_file: str = None, model_from_huggingface: str = None, random_seed: int = 765, batch_size: int = 32, num_examples: int = 5, num_examples_offset: int = 0)

Bases: ModelArgs, DatasetArgs

batch_size: int = 32

num_examples: int = 5

num_examples_offset: int = 0

random_seed: int = 765

ListThingsCommand class

class textattack.commands.list_things_command.ListThingsCommand

Bases: TextAttackCommand

The list module:

List default things in textattack.

static register_subcommand(main_parser: ArgumentParser)

run(args)

static things()

PeekDatasetCommand class

class textattack.commands.peek_dataset_command.PeekDatasetCommand

Bases: TextAttackCommand

The peek dataset module:

Takes a peek into a dataset in textattack.

static register_subcommand(main_parser: ArgumentParser)

run(args)

TextAttack CLI main class

textattack.commands.textattack_cli.main()

class textattack.commands.textattack_command.TextAttackCommand

Bases: ABC

abstract static register_subcommand(parser)

abstract run()

TrainModelCommand class

class textattack.commands.train_model_command.TrainModelCommand

Bases: TextAttackCommand

The TextAttack train module:

A command line parser to train a model from user specifications.

static register_subcommand(main_parser: ArgumentParser)

run(args)