View source on GitHub |
This class exports the EvalSavedModel in the end.
tfma.exporter.FinalExporter(
*args, **kwargs
)
This class performs a single export in the end of training.
Attributes | |
---|---|
name
|
Directory name.
A directory name under the export base directory where exports of
this type are written. Should not be |
Methods
export
export(
estimator: tf_estimator.Estimator,
export_path: str,
checkpoint_path: Optional[str],
eval_result: Optional[bytes],
is_the_final_export: bool
) -> Optional[bytes]
Exports the given Estimator
to a specific format.
Args | |
---|---|
estimator
|
the Estimator to export.
|
export_path
|
A string containing a directory where to write the export. |
checkpoint_path
|
The checkpoint path to export. |
eval_result
|
The output of Estimator.evaluate on this checkpoint.
|
is_the_final_export
|
This boolean is True when this is an export in the
end of training. It is False for the intermediate exports during the
training. When passing Exporter to tf.estimator.train_and_evaluate
is_the_final_export is always False if TrainSpec.max_steps is
None .
|
Returns | |
---|---|
The string path to the exported directory or None if export is skipped.
|