Specify a working model for a study variable y
Usage
y_model(formula, engine = c("glm", "tree", "forest", "boost"), family = NULL)
Arguments
- formula
full formula, e.g. income ~ sex + age_g.
- engine
"glm", "tree" (rpart), "forest" (ranger) or "boost" (xgboost).
- family
for engine = "glm": "gaussian", "binomial" or "poisson".
For tree/forest, regression vs classification is inferred from y.
Value
a model specification list.
Examples
y_model(income ~ age + sex, engine = "glm")
#> $formula
#> income ~ age + sex
#> <environment: 0x55aaee692d00>
#>
#> $engine
#> [1] "glm"
#>
#> $family
#> NULL
#>
