save_model_tf: Save/Load models using SavedModel format

Description Usage Arguments See Also

View source: R/model-persistence.R

Description

Save/Load models using SavedModel format

Usage

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
save_model_tf(
  object,
  filepath,
  overwrite = TRUE,
  include_optimizer = TRUE,
  signatures = NULL,
  options = NULL
)

load_model_tf(filepath, custom_objects = NULL, compile = TRUE)

Arguments

object

Model object to save

filepath

File path

overwrite

Overwrite existing file if necessary

include_optimizer

If TRUE, save optimizer's state.

signatures

Signatures to save with the SavedModel. Please see the signatures argument in tf$saved_model$save for details.

options

Optional tf$saved_model$SaveOptions object that specifies options for saving to SavedModel

custom_objects

Mapping class names (or function names) of custom (non-Keras) objects to class/functions (for example, custom metrics or custom loss functions).

compile

Whether to compile the model after loading.

See Also

Other model persistence: get_weights(), model_to_json(), model_to_yaml(), save_model_hdf5(), save_model_weights_hdf5(), serialize_model()


dfalbel/keras documentation built on Nov. 27, 2019, 8:16 p.m.