tfx.components.testdata.module_file.trainer_module.trainer_fn

Build the estimator using the high level API.

tfx.components.testdata.module_file.trainer_module.trainer_fn(
    trainer_fn_args, schema
)

Args
trainer_fn_args	Holds args used to train the model as name/value pairs.
schema	Holds the schema of the training examples.

Returns
A dict of the following: estimator: The estimator that will be used for training and eval. train_spec: Spec for training. eval_spec: Spec for eval. eval_input_receiver_fn: Input function for eval.