при подгонке earth
для модели glm
можно передать аргументы в вызов glm
. Например:
mars_fit <- earth(formula = response ~ x1 + x2, data = sim_dat,
glm = list(family=binomial, control = list(maxit = 50)))
Использование caret
выглядит как
fit_control <- trainControl(method = "cv", number = 10)
mars_grid <- expand.grid(degree=1:2, nprune=2:10)
mars_fit <- train(factor(response)~x1+x2, method='earth', trControl = fit_control,
data=sim_dat, tuneGrid=mars_grid,
glm = list(control = list(maxit = 50)))
но список glm
не передается. Любой совет?
Редактировать 1:
Читая https://github.com/topepo/caret/issues/554 caret
, автор говорит, что он либо пойман в ...
, либо его нужно передать в сетку настройки. при пропуске через сетку настройки, поскольку glm
является списком, train
жалуется, что degree
и nprune
не принадлежат методу, что неверно.
Изменить 2:
Открыт https://github.com/topepo/caret/issues/1018