首页 > 解决方案 > 使用 Caret R 包拟合多个模型(回归训练)

问题描述

我想使用 Caret R 包将多个模型拟合到同一个数据集。

我的结果是定量连续的,而我的预测是定性的二元(不存在与存在)。每行对应一个人。

我运行自动特征选择方法,然后从训练数据集中打乱预测变量。下一步是将多个模型拟合到同一个数据集。但是什么模型(methodList)通常适合我的数据?

标签: rmachine-learningr-caret

解决方案


查看模型信息,特别是他们的 $type 是否有“回归”

这至少应该是一个好的开始。如果你有 purrr,像这样,为简单起见,sapply如果没有,则等效调用:

mi <- getModelInfo()
ms %>% purrr::map_lgl( ~ "Regression" %in% .x$type )

推荐阅读