multiml.task.basic.sklean_pipeline module

SkleanPipelineTask module.

class multiml.task.basic.sklean_pipeline.SkleanPipelineTask(phases=None, input_var_names=None, output_var_names=None, save_var_names=None, pred_var_names=None, true_var_names=None, var_names=None, model=None, model_args=None, optimizer=None, optimizer_args=None, scheduler=None, scheduler_args=None, loss=None, loss_args=None, max_patience=None, loss_weights=None, load_weights=False, save_weights=False, metrics=None, num_epochs=10, batch_size=64, num_workers=0, verbose=None, **kwargs)

Bases: MLBaseTask

Wrapper task to process sklean object.

execute()

Execute fit.