nlp_architect.procedures.transformers package

Submodules

nlp_architect.procedures.transformers.base module

nlp_architect.procedures.transformers.base.create_base_args(parser: argparse.ArgumentParser, model_types=None)[source]: Add base arguments for Transformers based models

nlp_architect.procedures.transformers.base.inference_args(parser: argparse.ArgumentParser)[source]: Add inference specific arguments for Transoformer based models

nlp_architect.procedures.transformers.base.train_args(parser: argparse.ArgumentParser, models_family=None)[source]: Add training specific arguments for Transformer based models

nlp_architect.procedures.transformers.glue module

class nlp_architect.procedures.transformers.glue.TransformerGlueRun(parser=None)[source]

Bases: nlp_architect.procedures.procedure.Procedure

static add_arguments(parser: argparse.ArgumentParser)[source]

static run_procedure(args)[source]

class nlp_architect.procedures.transformers.glue.TransformerGlueTrain(parser=None)[source]

Bases: nlp_architect.procedures.procedure.Procedure

static add_arguments(parser: argparse.ArgumentParser)[source]

static run_procedure(args)[source]

nlp_architect.procedures.transformers.glue.add_glue_args(parser: argparse.ArgumentParser)[source]

nlp_architect.procedures.transformers.glue.add_glue_inference_args(parser: argparse.ArgumentParser)[source]

nlp_architect.procedures.transformers.glue.do_inference(args)[source]

nlp_architect.procedures.transformers.glue.do_training(args)[source]

nlp_architect.procedures.transformers.seq_tag module

class nlp_architect.procedures.transformers.seq_tag.TransformerTokenClsRun(parser=None)[source]

Bases: nlp_architect.procedures.procedure.Procedure

static add_arguments(parser: argparse.ArgumentParser)[source]

static run_procedure(args)[source]

class nlp_architect.procedures.transformers.seq_tag.TransformerTokenClsTrain(parser=None)[source]

Bases: nlp_architect.procedures.procedure.Procedure

static add_arguments(parser: argparse.ArgumentParser)[source]

static run_procedure(args)[source]

nlp_architect.procedures.transformers.seq_tag.do_inference(args)[source]

nlp_architect.procedures.transformers.seq_tag.do_training(args)[source]

nlp_architect.procedures.transformers.seq_tag.process_inference_input(input_file)[source]

Module contents