|View source on GitHub|
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.
run_fn(...): Train the model based on given args.
trainer_fn(...): Build the estimator using the high level API.