Save and load Ruta models

Functions to save a trained or untrained Ruta learner into a file and load it

save_as(learner, file = paste0(substitute(learner), ".tar.gz"), dir,
  compression = "gzip")

load_from(file)

Arguments

learner	The "ruta_autoencoder" object to be saved
file	In save, filename with extension (usually .tar.gz) where the object will be saved. In load, path to the saved model
dir	Directory where to save the file. Use "." to save in the current working directory or tempdir() to use a temporary one
compression	Type of compression to be used, for R function tar

Value

save_as returns the filename where the model has been saved, load_from returns the loaded model as a "ruta_autoencoder" object

Examples

library(purrr)

x <- as.matrix(iris[, 1:4])
# Save a trained model
saved_file <-
  autoencoder(2) %>%
  train(x) %>%
  save_as("my_model.tar.gz", dir = tempdir())

# Load and use the model
encoded <- load_from(saved_file) %>% encode(x)
#> Loading weights from /tmp/RtmpwJrBqN/ruta/weights.hdf5