Add a column with predicted values from a regression model

The regression output comes from stats::lm. The effect sizes are calculated by heplots::etasq. The variance inflation is calculated by car::vif.

Usage

add_model(
  data,
  col,
  categorical,
  metric,
  interactions = NULL,
  labels = TRUE,
  clean = TRUE,
  ...
)

Arguments

data: A tibble.
col: The target column holding metric values.
categorical: A tidy column selection holding categorical variables.
metric: A tidy column selection holding metric variables.
interactions: A vector of interaction effects to calculate. Each interaction effect should be provided as multiplication of the variables. The interaction effect can be provided as character value (e.g. c("sd_gender * adopter")) or as unquoted column names (e.g. c(sd_gender * adopter)).
labels: If TRUE (default) extracts labels from the attributes, see codebook.
clean: Prepare data by data_clean.
...: Placeholder to allow calling the method with unused parameters from effect_metrics.

Value

The input tibble with one additional column. The new column name is derived from the target column, prefixed with "prd_". The new column will have an attribute "lm.fit" with the fit model.

Examples

library(volker)
data <- filter(volker::chatgpt, sd_gender != "diverse")

data <- data |>
  add_model(use_work, categorical = c(sd_gender, adopter), metric = sd_age)