autokeras (version 1.0.1)

save_model: Save Model

Description

Save the AutoKeras Model. Make sure that `directory` and `name` values are set when creating the model, and that the `directory` is not a temporary folder.

Usage

save_model(autokeras_model, filename)

Arguments

autokeras_model

: A trained AutokerasModel instance.

filename

: A character string naming a file to save the model.

Value

None

Examples

# NOT RUN {
library("keras")

# use the MNIST dataset as an example
mnist <- dataset_mnist()
c(x_train, y_train) %<-% mnist$train
c(x_test, y_test) %<-% mnist$test

library("autokeras")

# Initialize the image classifier
clf <- model_image_classifier(
  directory = ".",
  name = "autokeras_mnist",
  max_trials = 10
) %>% # It tries 10 different models
  fit(x_train, y_train) # Feed the image classifier with training data

# Predict with the best model
(predicted_y <- clf %>% predict(x_test))

# Evaluate the best model with testing data
clf %>% evaluate(x_test, y_test)

# Get the best trained Keras model, to work with the keras R library
export_model(clf)

# Save the AutoKeras model.
# Make sure that `directory` and `name` were set when creating the model.
save_model(clf, "my_model.pkl")

# And load it again
new_clf <- load_model("my_model.pkl")
# }
# NOT RUN {
# }