/
model_get_xlevels.R
83 lines (70 loc) · 1.84 KB
/
model_get_xlevels.R
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
#' Get xlevels used in the model
#'
#' @param model a model object
#' @export
#' @family model_helpers
#' @examples
#' lm(hp ~ mpg + factor(cyl), mtcars) %>%
#' model_get_xlevels()
model_get_xlevels <- function(model) {
UseMethod("model_get_xlevels")
}
#' @export
#' @rdname model_get_xlevels
model_get_xlevels.default <- function(model) {
xlevels <- tryCatch(
model %>% purrr::chuck("xlevels"),
error = function(e) {
NULL # nocov
}
)
if (is.null(xlevels)) {
xlevels <- tryCatch(
stats::.getXlevels(
stats::terms(model),
model %>% model_get_model_frame()
),
error = function(e) {
NULL # nocov
}
)
}
xlevels %>% .add_xlevels_for_logical_variables(model)
}
.add_xlevels_for_logical_variables <- function(xlevels, model) {
log_vars <- model %>%
model_list_variables() %>%
dplyr::filter(.data$var_class == "logical") %>%
purrr::pluck("variable")
for (v in setdiff(log_vars, names(xlevels)))
xlevels[[v]] <- c("FALSE", "TRUE")
xlevels
}
#' @export
#' @rdname model_get_xlevels
model_get_xlevels.lmerMod <- function(model) {
stats::model.frame(model) %>%
lapply(levels) %>%
purrr::compact() %>% # keep only not null
.add_xlevels_for_logical_variables(model)
}
#' @export
#' @rdname model_get_xlevels
model_get_xlevels.glmerMod <- model_get_xlevels.lmerMod
#' @export
#' @rdname model_get_xlevels
model_get_xlevels.felm <- model_get_xlevels.lmerMod
#' @export
#' @rdname model_get_xlevels
model_get_xlevels.brmsfit <- model_get_xlevels.lmerMod
#' @export
#' @rdname model_get_xlevels
model_get_xlevels.glmmTMB <- model_get_xlevels.lmerMod
#' @export
#' @rdname model_get_xlevels
model_get_xlevels.plm <- model_get_xlevels.lmerMod
#' @export
#' @rdname model_get_xlevels
model_get_xlevels.model_fit <- function(model) {
model_get_xlevels(model$fit)
}