multiml.task.basic.modules.connection_model module

class multiml.task.basic.modules.connection_model.ConnectionModel(models=None, input_var_index=None, output_var_index=None, *args, **kwargs)

Bases: object

__init__(models=None, input_var_index=None, output_var_index=None, *args, **kwargs)

Connecting ML models differentially.

This ConnectionModel is usually build inside ModelConnectionTask to set proper indexes automatically. Inputs data for call() or forward() are assumed to be list format.

Parameters:

  • models (list) – list of compiled models.

  • input_var_index (list) – list of index to input_var_names.

  • *args – Variable length argument list.

  • **kwargs – Arbitrary keyword arguments.