Join the SIG TFX-Addons community and help make TFX even better!

tfx.types.standard_component_specs.TrainerSpec

Trainer component spec.

Inherits From: ComponentSpec

**kwargs Any inputs, outputs and execution parameters for this instance of the component spec.

Methods

__eq__

View source

Return self==value.

INPUTS

{
 'base_model': ChannelParameter(type: <class 'tfx.types.standard_artifacts.Model'>),
 'examples': ChannelParameter(type: <class 'tfx.types.standard_artifacts.Examples'>),
 'hyperparameters': ChannelParameter(type: <class 'tfx.types.standard_artifacts.HyperParameters'>),
 'schema': ChannelParameter(type: <class 'tfx.types.standard_artifacts.Schema'>),
 'transform_graph': ChannelParameter(type: <class 'tfx.types.standard_artifacts.TransformGraph'>)
}

OUTPUTS

{
 'model': ChannelParameter(type: <class 'tfx.types.standard_artifacts.Model'>),
 'model_run': ChannelParameter(type: <class 'tfx.types.standard_artifacts.ModelRun'>)
}

PARAMETERS

{
 'custom_config': ExecutionParameter(type: (<class 'str'>, <class 'str'>), optional: True),
 'eval_args': ExecutionParameter(type: <class 'tfx.proto.trainer_pb2.EvalArgs'>, optional: False),
 'module_file': ExecutionParameter(type: (<class 'str'>, <class 'str'>), optional: True),
 'module_path': ExecutionParameter(type: (<class 'str'>, <class 'str'>), optional: True),
 'run_fn': ExecutionParameter(type: (<class 'str'>, <class 'str'>), optional: True),
 'train_args': ExecutionParameter(type: <class 'tfx.proto.trainer_pb2.TrainArgs'>, optional: False),
 'trainer_fn': ExecutionParameter(type: (<class 'str'>, <class 'str'>), optional: True)
}