如何提取线性模型回归系数时，使用M5的插入程序包的训练函数？

Question

我已经使用caret包的训练函数构建了一个M5 regression模型树。我正在尝试提取树的每个模型的linear model coefficients。

最终的模型称为m5tune。我试着去找m5tune$finalModel，但是我没有找到。coeff()函数似乎也不起作用...

Answer 1

使用这种方法，您可以获得相关系数。请注意，这些数据仅具有代表性。

library(Rweka)
iris1<-as.data.frame(iris)
set.seed(456)
mytrain<-createDataPartition(iris$Sepal.Length,p=0.8,list=F)
trainme<-iris[mytrain,]
mymodel<-train(Sepal.Length~.,trainme,trControl=trainControl(method="cv",number=5),method="M5",
               metric="MAE")

简单地调用它就可以得到相关系数。

summary(mymodel)