使用 tidymodels 的 GLM 系列

Question

我正在嘗試將tidymodels package 用於 GLM，並希望使用 Gamma 或 Poisson 分布。

使用glm我會使用類似下面的東西

# using glm
mdl <- glm(data = data, y ~ x, family = Gamma(link = "inverse"))
mdl <- glm(data = data, y ~ x, family = poisson(link = "log"))

# using glmnet
library(glmnet)
mdl <- glmnet(data$x, data$y, family = Gamma(link = "inverse"))
mdl <- glmnet(data$x, data$y, family = poisson(link = "log"))

如何使用tidymodels實現相同的目標？ 請注意，我正在嘗試進行回歸而不是可以使用parsnip::logistic_reg() 。

我在tidymodels上找到了一篇關於Generalized Linear Models的文章，屬於embed package但沒有說明如何指定族。

我希望有類似的東西（這不起作用，因為linear_reg沒有參數family或link ， set_engine也不支持glm線性回歸模式）

mdl <- linear_reg(mode = "regression", family = "gamma", link = "inverse") %>% set_engine("glm") # or glmnet

Answer 1

這比預期的要容易：

mdl <- linear_reg(mode = "regression") %>%
  set_engine("glmnet", family = "gamma")

# or 
mdl <- linear_reg(mode = "regression") %>%
  set_engine("glmnet", family = Gamma(link = "inverse"))