Как передать аргумент управления GLM для Земли с помощью каретки (maxit) - PullRequest
0 голосов
/ 17 марта 2019

при подгонке 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

1 Ответ

0 голосов
/ 26 марта 2019

проблема решена в этом коммите: https://github.com/topepo/caret/commit/2ce2cf4c5889791b7dbca5d8896fcc6dc0d0bcfc

...