multiml.task.basic.ml_env module

class multiml.task.basic.ml_env.MLEnv(model=None, optimizer=None, scheduler=None, loss=None, loss_weights=None, multi_inputs=None, multi_outputs=None, multi_loss=None)

Bases: object

Data class to store compiled ML objects.

__init__(model=None, optimizer=None, scheduler=None, loss=None, loss_weights=None, multi_inputs=None, multi_outputs=None, multi_loss=None)

Initialize MLEnv.

clear()

property model

Returns model.

property optimizer

Returns optimizer.

property scheduler

Returns scheduler.

property loss

Returns loss.

property loss_weights

Returns loss_weights.

property multi_inputs

Returns multi_inputs.

property multi_outputs

Returns multi_outputs.

property multi_loss

Returns multi_loss.

show_info()

Print information.

validate(phase)

Validate environment for given phase.