Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tf.estimator.Exporter

TensorFlow 1 version View source on GitHub

A class representing a type of model export.

Attributes:

  • name: Directory name.

    A directory name under the export base directory where exports of this type are written. Should not be None nor empty.

Methods

export

View source

export(
    estimator, export_path, checkpoint_path, eval_result, is_the_final_export
)

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.