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/keras/estimator/__init__.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.