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

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/ and examples/chicago_taxi/


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

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