使用 caret 包时如何检索特定于模型的信息?

机器算法验证 r 插入符号
2022-04-07 07:15:36

CARET 是一个方便的工具,它包含了数十种建模方法。CARET 允许用户以通用方式访问性能统计数据,例如 RMSE、混淆矩阵等,而无需回忆各种建模方法的使用。

但是,某些模型具有独特的报告结果。例如,使用增强树模型,可以将样本内样本外的错误率作为树数的函数进行比较,以测试过度拟合。自然,此报告对于不使用树的其他方法没有意义。

有没有办法可以使用 CARET 访问这些特定于模型的报告(也许通过访问直接生成的建模对象)?

1个回答

中的train函数caret返回一个 S3 列表对象。此列表中的一项称为finalModel这个项目只是它的原生形式的适合对象。因此,如果您使用 gbm 训练 gbmtrain并将其存储在名为 的对象中my.fitsummary(my.fit$finalModel)则将使用包中的summary.gbm方法gbm并向您显示相对变量的重要性。