RSVP for your your local TensorFlow Everywhere event today!

Module: tfx.components.testdata.module_file.trainer_module

Python source file include taxi pipeline functions and necesasry utils.

For a TFX pipeline to successfully run, a preprocessing_fn and a _build_estimator function needs to be provided. This file contains both.

This file is equivalent to examples/chicago_taxi/trainer/model.py and examples/chicago_taxi/preprocess.py.

Functions

run_fn(...): Train the model based on given args.

trainer_fn(...): Build the estimator using the high level API.