Module: tfx.components.trainer.fn_args_utils

View source on GitHub

FnArgs for passing information to UDF.

Classes

class FnArgs: FnArgs(working_dir, train_files, eval_files, train_steps, eval_steps, schema_path, transform_graph_path)

Functions

get_common_fn_args(...): Get common args of training and tuning.