Установите модель без перехвата в каретке
В R я указываю модель без перехвата следующим образом:
data(iris)
lmFit <- lm(Sepal.Length ~ 0 + Petal.Length + Petal.Width, data=iris)
> round(coef(lmFit),2)
Petal.Length Petal.Width
2.86 -4.48
Однако, если я вхожу в ту же модель с кареткой, результирующая модель включает в себя перехват:
library(caret)
caret_lmFit <- train(Sepal.Length~0+Petal.Length+Petal.Width, data=iris, "lm")
> round(coef(caret_lmFit$finalModel),2)
(Intercept) Petal.Length Petal.Width
4.19 0.54 -0.32
Как сообщить caret::train
исключить термин перехвата?
Ответы
Ответ 1
Как обсуждалось в связанном вопросе SO fooobar.com/questions/374632/..., это работает в caret v6.0.76 (И ответ на трассировку выше больше не работает с рефакторингом кода в каретка):
caret_lmFit <- train(Sepal.Length~0+Petal.Length+Petal.Width, data=iris, "lm",
tuneGrid = expand.grid(intercept = FALSE))
> caret_lmFit$finalModel
Call:
lm(formula = .outcome ~ 0 + ., data = dat)
Coefficients:
Petal.Length Petal.Width
2.856 -4.479
Ответ 2
@rcs уже сказал вам, какая строка, в какой функции вам нужно изменить.
Просто используйте trace
, чтобы изменить эту функцию:
trace(caret::createModel,
quote(modFormula <- as.formula(".outcome ~ .-1")), at=5, print=FALSE)
caret_lmFit <- train(Sepal.Length~0+Petal.Length+Petal.Width, data=iris, "lm")
round(coef(caret_lmFit$finalModel),2)
#Petal.Length Petal.Width
# 2.86 -4.48
untrace(caret::createModel)
Однако я не использую каретку. Возможны непредвиденные последствия. Также часто не рекомендуется исключать перехват из модели.