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

tfx.components.trainer.fn_args_utils.FnArgs

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

working_dir Working dir.
train_files A list of patterns for train files.
eval_files A list of patterns for eval files.
train_steps Number of train steps.
eval_steps Number of eval steps.
schema_path A single uri for schema file. Will be None if not specified.
schema_file Deprecated, use schema_path instead.
transform_graph_path An optional single uri for transform graph produced by TFT. Will be None if not specified.
transform_output Deprecated, use transform_graph_path instead.'
data_accessor Contains factories that can create tf.data.Datasets or other means to access the train/eval data. They provide a uniform way of accessing data, regardless of how the data is stored on disk.
serving_model_dir A single uri for the output directory of the serving model.
eval_model_dir A single uri for the output directory of the eval model. Note that this is estimator only, Keras doesn't require it for TFMA.
model_run_dir A single uri for the output directory of model training related files.
base_model An optional base model path that will be used for this training.
hyperparameters An optional kerastuner.HyperParameters config.
custom_config An optional dictionary passed to the component.

Methods

__eq__

Method generated by attrs for class FnArgs.

__ge__

Method generated by attrs for class FnArgs.

__gt__

Method generated by attrs for class FnArgs.

__le__

Method generated by attrs for class FnArgs.

__lt__

Method generated by attrs for class FnArgs.

__ne__

Method generated by attrs for class FnArgs.