mlr3 - 是否可以从重采样结果中获得对训练数据的预测？

Question

执行以下代码后，可以使用rr$predictions()[[1]]. 但是有可能获得关于训练分区的预测吗？

task = tsk("penguins")
learner = lrn("classif.rpart")
resampling = rsmp("holdout")
rr = resample(task, learner, resampling)

谢谢！

标签： mlr3

Answer 1

您需要将predict_sets学习者的字段设置为训练和测试，如下所示：

learner$predict_sets=c("test", "train")

保持其他一切不变并获得训练集预测

rr$prediction("train")