keras_model_to_estimator {tfestimators}

Keras Estimators

Description

Create an Estimator from a compiled Keras model

Usage

keras_model_to_estimator(
  keras_model = NULL,
  keras_model_path = NULL,
  custom_objects = NULL,
  model_dir = NULL,
  config = NULL
)

Arguments

keras_model

A keras model.

keras_model_path

Directory to a keras model on disk.

custom_objects

Dictionary for custom objects.

model_dir

Directory to save Estimator model parameters, graph and etc.

config

Configuration object.