multiml.task.basic.modules package

Submodules

• multiml.task.basic.modules.connection_model module
  • ConnectionModel
    • ConnectionModel.__init__()

Module contents

class multiml.task.basic.modules.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.