tf.keras.estimator.model_to_estimator

tf.keras.estimator.model_to_estimator(
    keras_model=None,
    keras_model_path=None,
    custom_objects=None,
    model_dir=None,
    config=None
)

Defined in tensorflow/python/estimator/keras.py.

Constructs an Estimator instance from given keras model.

For usage example, please see: Creating estimators from Keras Models.

Args:

  • keras_model: A compiled Keras model object. This argument is mutually exclusive with keras_model_path.
  • keras_model_path: Path to a compiled Keras model saved on disk, in HDF5 format, which can be generated with the save() method of a Keras model. This argument is mutually exclusive with keras_model.
  • custom_objects: Dictionary for custom objects.
  • model_dir: Directory to save Estimator model parameters, graph, summary files for TensorBoard, etc.
  • config: RunConfig to config Estimator.

Returns:

An Estimator from given keras model.

Raises:

  • ValueError: if neither keras_model nor keras_model_path was given.
  • ValueError: if both keras_model and keras_model_path was given.
  • ValueError: if the keras_model_path is a GCS URI.
  • ValueError: if keras_model has not been compiled.