mlr3在哪里保存最终模型?

问题描述

训练了学习者-mlr3之后,learner$train(data)在哪里保存最终模型? “最终模型”是指类似于以下代码生成的列表:

model <- xgboost::xgb.train(data = data_train,max.depth = 8,nthread = 2,nrounds = 15,verbose = 0)  

有没有办法提取此列表/对象?


task <- TaskRegr$new("data",data,"y")
learner <- lrn("regr.xgboost")
preprocess <- po("scale",param_vals = list(center = TRUE,scale = TRUE))
pp <- preprocess %>>% learner
gg<- GraphLearner$new(pp)
gg$train(task)

解决方法

在xgboost中,“模型”存储为:

model <- xgboost::xgb.train(data = data_train,max.depth = 8,nthread = 2,nrounds = 15,verbose = 0)

在MLR3中,使用以下方法进行训练:

task <- TaskRegr$new("data",data,"y")
learner <- lrn("regr.xgboost")
preprocess <- po("scale",param_vals = list(center = TRUE,scale = TRUE))
pp <- preprocess %>>% learner
gg<- GraphLearner$new(pp)
gg$train(task)

相当于“模型”的存储为

gg$model$regr.xgboost$model

相关问答

Selenium Web驱动程序和Java。元素在(x,y)点处不可单击。其...
Python-如何使用点“。” 访问字典成员?
Java 字符串是不可变的。到底是什么意思?
Java中的“ final”关键字如何工作?(我仍然可以修改对象。...
“loop:”在Java代码中。这是什么,为什么要编译?
java.lang.ClassNotFoundException:sun.jdbc.odbc.JdbcOdbc...