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

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
)

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

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.