lightning_template.utils.mixin.split_name_mixin

Classes

SplitNameMixin

Module Contents

class lightning_template.utils.mixin.split_name_mixin.SplitNameMixin

TrainSplit = 'fit'

ValidateSplit = 'val'

TestSplit = 'test'

PredictSplit = 'predict'

SplitNameMap

split_names

_get_split_names(stage=None)

setup(stage=None)

static substitute_split_name(cur_cfg, split_prefix, split_attr, split_attr_split_str, split_name_map, split_name, *args, **kwargs)

get_split_config(config)