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)

Add base arguments for Transformers based models

nlp_architect.procedures.transformers.base.inference_args(parser: argparse.ArgumentParser)

Add inference specific arguments for Transoformer based models

nlp_architect.procedures.transformers.base.train_args(parser: argparse.ArgumentParser, models_family=None)

Add training specific arguments for Transformer based models

nlp_architect.procedures.transformers.glue module

class nlp_architect.procedures.transformers.glue.TransformerGlueRun(parser=None)

Bases: nlp_architect.procedures.procedure.Procedure

static add_arguments(parser: argparse.ArgumentParser)

static run_procedure(args)

class nlp_architect.procedures.transformers.glue.TransformerGlueTrain(parser=None)

Bases: nlp_architect.procedures.procedure.Procedure

static add_arguments(parser: argparse.ArgumentParser)

static run_procedure(args)

nlp_architect.procedures.transformers.glue.add_glue_args(parser: argparse.ArgumentParser)

nlp_architect.procedures.transformers.glue.add_glue_inference_args(parser: argparse.ArgumentParser)

nlp_architect.procedures.transformers.glue.do_inference(args)

nlp_architect.procedures.transformers.glue.do_training(args)

nlp_architect.procedures.transformers.seq_tag module

class nlp_architect.procedures.transformers.seq_tag.TransformerTokenClsRun(parser=None)

Bases: nlp_architect.procedures.procedure.Procedure

static add_arguments(parser: argparse.ArgumentParser)

static run_procedure(args)

class nlp_architect.procedures.transformers.seq_tag.TransformerTokenClsTrain(parser=None)

Bases: nlp_architect.procedures.procedure.Procedure

static add_arguments(parser: argparse.ArgumentParser)

static run_procedure(args)

nlp_architect.procedures.transformers.seq_tag.do_inference(args)

nlp_architect.procedures.transformers.seq_tag.do_training(args)

nlp_architect.procedures.transformers.seq_tag.process_inference_input(input_file)

Module contents