hyrax.verbs.train

hyrax.verbs.train#

Attributes#

logger

Classes#

Train

Train verb

Module Contents#

logger[source]#
class Train(config)[source]#

Bases: hyrax.verbs.verb_registry.Verb

Train verb

__init__()[source]#

Overall initialization for all verbs that saves the config

cli_name = 'train'[source]#
add_parser_kwargs[source]#
description = 'Train a model using provided data.'[source]#
REQUIRED_DATA_GROUPS = ('train',)[source]#
OPTIONAL_DATA_GROUPS = ('validate', 'test')[source]#
static setup_parser(parser)[source]#

We don’t need any parser setup for CLI opts

run_cli(args=None)[source]#

CLI stub for Train verb

run()[source]#

Run the training process for the configured model and data loader. Returns the trained model.

static _log_params(config, results_dir)[source]#

Log the various parameters to mlflow from the config file.

Parameters:

  • config (dict) – The main configuration dictionary

  • results_dir (str) – The full path to the results sub-directory