【发布时间】:2023-02-07 02:54:43
【问题描述】:
为了模仿 caret 如何执行 RFE 并选择产生最低 RMSE 的特征,建议使用存档。
我正在使用 AutoFSelector 并使用以下代码进行嵌套重采样:
ARMSS<-read.csv("Index ARMSS Proteomics Final.csv", row.names=1)
set.seed(123, "L'Ecuyer")
task = as_task_regr(ARMSS, target = "Index.ARMSS")
learner = lrn("regr.ranger", importance = "impurity")
set_threads(learner, n = 8)
resampling_inner = rsmp("cv", folds = 7)
measure = msr("regr.rmse")
terminator = trm("none")
at = AutoFSelector$new(
learner = learner,
resampling = resampling_inner,
measure = measure,
terminator = terminator,
fselect = fs("rfe", n_features = 1, feature_fraction = 0.5, recursive = FALSE),
store_models = TRUE)
resampling_outer = rsmp("repeated_cv", folds = 10, repeats = 10)
rr = resample(task, at, resampling_outer, store_models = TRUE)
我是否应该使用 extract_inner_fselect_archives() 命令来识别具有最小 RMSE 和所选特征的每次迭代,然后重新运行上面的代码并更改 n_features 参数?如何协调迭代中特征数量和/或所选特征的差异?
【问题讨论】:
标签: mlr3