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

Module: tfx.components.trainer.fn_args_utils

FnArgs for passing information to UDF.

Classes

class DataAccessor: DataAccessor(tf_dataset_factory, record_batch_factory)

class FnArgs: Args to pass to user defined training/tuning function(s).

Functions

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