Ejemplo n.º 1
0
def _add_training_arguments(
        parser: argparse.ArgumentParser) -> argparse._ArgumentGroup:
    train_arguments = parser.add_argument_group("Train Arguments")
    add_config_param(train_arguments)
    add_domain_param(train_arguments)
    add_out_param(train_arguments,
                  help_text="Directory where your models should be stored.")
    add_augmentation_param(train_arguments)
    add_debug_plots_param(train_arguments)

    return train_arguments
Ejemplo n.º 2
0
def set_interactive_core_arguments(parser: argparse.ArgumentParser):
    add_model_param(parser)
    add_config_param(parser)
    add_domain_param(parser)
    add_stories_param(parser)
    add_out_param(parser)

    add_augmentation_param(parser)
    add_debug_plots_param(parser)
    add_dump_stories_param(parser)

    add_skip_visualization_param(parser)

    add_server_arguments(parser)
Ejemplo n.º 3
0
def set_interactive_core_arguments(parser: argparse.ArgumentParser):
    add_model_param(parser, model_name="Rasa Core", default=None)
    add_stories_param(parser)

    add_skip_visualization_param(parser)

    add_endpoint_param(
        parser,
        help_text=
        "Configuration file for the model server and the connectors as a yml file.",
    )

    train_arguments = parser.add_argument_group("Train Arguments")
    add_config_param(train_arguments)
    add_domain_param(train_arguments)
    add_out_param(train_arguments)
    add_augmentation_param(train_arguments)
    add_debug_plots_param(train_arguments)
    add_dump_stories_param(train_arguments)
Ejemplo n.º 4
0
def set_interactive_arguments(parser: argparse.ArgumentParser):
    add_model_param(parser, default=None)
    add_data_param(parser)

    add_skip_visualization_param(parser)

    add_endpoint_param(
        parser,
        help_text=
        "Configuration file for the model server and the connectors as a yml file.",
    )

    train_arguments = parser.add_argument_group("Train Arguments")
    add_config_param(train_arguments)
    add_domain_param(train_arguments)
    add_out_param(train_arguments,
                  help_text="Directory where your models should be stored.")
    add_augmentation_param(train_arguments)
    add_debug_plots_param(train_arguments)
    add_dump_stories_param(train_arguments)
    add_force_param(train_arguments)