保存和加载mlr模型

时间:2018-09-02 21:44:00

标签: r multilabel-classification mlr

使用以下设置,我已经能够使用mlr库成功生成模型以协助我的多标签分配

scene.task = makeMultilabelTask(data = training.data, target = labels)
lrn.br = makeLearner("classif.rpart", predict.type = "prob")
lrn.br = makeMultilabelNestedStackingWrapper(lrn.br)
mdl.lrn <- train(lrn.br, scene.task)

我要使用以下代码保存模型,

save(mdl.lrn, file = "mdl.rda")

所保存的模型大小刚好超过1GB。

我尝试使用以下行加载该保存的模型,

load("mdl.rda")

但是,当我收到读为Error: cannot allocate vector of size ### Kb的消息时,我的内存空间不断用完。我相信我当前的内存量将允许加载模型,因为它足以生成模型。

有人能保存和加载mlr生成的模型以供重用吗?如果是这样,您能否推荐或确认保存和加载模型的方法?

0 个答案:

没有答案