hyrax.verbs.create_splits

Attributes

logger

Classes

CreateSplits

Create and persist reproducible dataset splits.

Module Contents

logger

class CreateSplits(config)

Bases: hyrax.verbs.verb_registry.Verb

Create and persist reproducible dataset splits.

__init__()

Overall initialization for all verbs that saves the config

cli_name = 'create_splits'

add_parser_kwargs

description = 'Compute and persist dataset splits for reproducible training workflows.'

REQUIRED_DATA_GROUPS = ()

OPTIONAL_DATA_GROUPS = ()

static setup_parser(parser)

No additional CLI options needed.

run_cli(args=None)

CLI stub for CreateSplits verb.

run()

Compute dataset splits and write them to a results directory.

Reads the [split] and [balance] config tables to determine how to partition each data group, then persists .npz index files and a split_config.toml under a timestamped *-splits-* results directory. Subsequent verbs (train, infer, test) can point at this directory to reuse the same split without recomputing it.

Returns:

The populated dataset providers, keyed by group name.

Return type:

dict[str, DataProvider]