Как извлечь коэффициенты регрессии линейных моделей при использовании M5 с функцией поезда пакета каретки? - PullRequest
0 голосов
/ 18 декабря 2018

Я построил дерево моделей M5 , используя функцию поезда из пакета caret.Я пытаюсь извлечь linear model coefficients для каждой модели дерева.

Окончательная модель называется m5tune.Я пытался заглянуть в m5tune$finalModel, но я не нашел его.Функция coeff(), похоже, тоже не работает ...

1 Ответ

0 голосов
/ 18 декабря 2018

Используя этот подход, вы можете получить коэффициент корреляции.Пожалуйста, данные являются только репрезентативными.

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)
...