![]() |
Args to pass to user defined training/tuning function(s).
tfx.components.trainer.fn_args_utils.FnArgs(
working_dir=attr_dict['working_dir'].default,
train_files=attr_dict['train_files'].default,
eval_files=attr_dict['eval_files'].default,
train_steps=attr_dict['train_steps'].default,
eval_steps=attr_dict['eval_steps'].default,
schema_path=attr_dict['schema_path'].default,
schema_file=attr_dict['schema_file'].default,
transform_graph_path=attr_dict['transform_graph_path'].default,
transform_output=attr_dict['transform_output'].default,
data_accessor=attr_dict['data_accessor'].default,
serving_model_dir=attr_dict['serving_model_dir'].default,
eval_model_dir=attr_dict['eval_model_dir'].default,
model_run_dir=attr_dict['model_run_dir'].default,
base_model=attr_dict['base_model'].default,
hyperparameters=attr_dict['hyperparameters'].default,
custom_config=attr_dict['custom_config'].default
)
Attributes | |
---|---|
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__
__eq__(
other
)
Method generated by attrs for class FnArgs.
__ge__
__ge__(
other
)
Method generated by attrs for class FnArgs.
__gt__
__gt__(
other
)
Method generated by attrs for class FnArgs.
__le__
__le__(
other
)
Method generated by attrs for class FnArgs.
__lt__
__lt__(
other
)
Method generated by attrs for class FnArgs.
__ne__
__ne__(
other
)
Method generated by attrs for class FnArgs.