使用mlr3对维持集进行预测

wn9m85ua 于 2023-02-14 发布在其他

关注(0)|答案(1)|浏览(125)

我使用的是mlr3包，我将一些行的角色设置为“holdout”，然后训练模型：

library(mlr3)

# train on iris
task = tsk("iris")
task$nrow
task$set_row_roles(130:150, "holdout")
learner = lrn("classif.rpart")
learner$train(task)

我怎么知道使用支持集来预测它们呢？

# predict on holdout
task$row_roles$holdout
## HOW TO PREDICT ON HOLDOUT SET ? 
# learner$predict()

来源：https://stackoverflow.com/questions/75409141/predict-on-holdout-set-using-mlr3

1条答案

按热度按时间

utugiqy61#

可以在$predict()方法中设置要获取预测的行ID。

library(mlr3)

task = tsk("iris")
task$nrow
#> [1] 150
task$set_row_roles(130:150, "holdout")
learner = lrn("classif.rpart")
learner$train(task)

learner$predict(task, row_ids = task$row_roles$holdout)
#> <PredictionClassif> for 21 observations:
#>     row_ids     truth   response
#>         130 virginica versicolor
#>         131 virginica  virginica
#>         132 virginica  virginica
#> ---                             
#>         148 virginica  virginica
#>         149 virginica  virginica
#>         150 virginica  virginica

创建于2023年2月10日，使用reprex v2.0.2

赞(0）回复(0）举报 2023-02-14

我来回答

使用mlr3对维持集进行预测

1条答案

相关问题

热门标签

最新问答