web-dev-qa-db-fra.com

Enregistrement et chargement d'un modèle dans R

Lorsque vous travaillez avec caret , comment puis-je enregistrer un modèle après l'entraînement et le charger plus tard (par exemple dans une autre session) pour la prédiction?

28

De nos jours, une meilleure solution consiste à utiliser saveRDS pour enregistrer et readRDS pour lire:

saveRDS(model, "model.rds")
my_model <- readRDS("model.rds")

Cela vous permet de choisir un nouveau nom pour l'objet (vous n'avez pas besoin de vous souvenir du nom que vous avez utilisé lorsque vous l'avez enregistré)

25
lbcommer

La syntaxe correcte serait d'utiliser:

save(model, file="model.Rdata")

Par la suite, il peut être chargé à l'aide de la commande load ().

14
Dan Ibanez

Le code suivant suppose que le nom de variable de votre modèle est "modèle":

save(model, "model.RData")

Cela enregistrera votre modèle en tant que "model.RData" dans le répertoire de travail actuel. Vous pouvez découvrir ce qu'est le répertoire de travail en émettant ce qui suit:

getwd()

Pour le recharger, assurez-vous que votre modèle est enregistré dans votre répertoire de travail et lancez:

load("model.RData")
10
Nick Evans