multiml.task.pytorch.pytorch_choice_block module

class multiml.task.pytorch.pytorch_choice_block.PytorchChoiceBlockTask(subtasks, **kwargs)

Bases: PytorchBaseTask

__init__(subtasks, **kwargs)

Parameters:

• subtasks (list) – list of task instances.

• **kwargs – Arbitrary keyword arguments.

build_model()

Build model.

property choice

get_input_true_data(phase)

Get input and true data.

Parameters:

phase (str) – data type (train, valid, test or None).

Returns:

(input, true) data for model.

Return type:

tuple

get_storegate_dataset(phase)

Returns storegate dataset.

get_submodel_names()

get_inputs()

get_submodel(i_models)