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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
|
#' @export
model_parameters.mediate <- function(model, ci = 0.95, exponentiate = FALSE, verbose = TRUE, ...) {
# Parameters, Estimate and CI
params <- insight::get_parameters(model)
# CI
params <- merge(params, ci(model, ci = ci), by = "Parameter", sort = FALSE)
params$CI <- NULL
# p-value
params <- merge(params, p_value(model), by = "Parameter", sort = FALSE)
# ==== Renaming
if (any(endsWith(params$Parameter, "(control)"))) {
params$Component <- gsub("(.*)\\((.*)\\)$", "\\2", params$Parameter)
}
# exponentiate coefficients and SE/CI, if requested
params <- .exponentiate_parameters(params, model, exponentiate)
attr(params, "object_name") <- insight::safe_deparse_symbol(substitute(model))
params <- .add_model_parameters_attributes(params, model, ci, exponentiate, verbose = verbose, ...)
class(params) <- c("parameters_model", "see_parameters_model", class(params))
params
}
#' @export
ci.mediate <- function(x, ci = 0.95, ...) {
info <- insight::model_info(x$model.y, verbose = FALSE)
alpha <- (1 + ci) / 2
if (info$is_linear && !x$INT) {
out <- data.frame(
Parameter = c("ACME", "ADE", "Total Effect", "Prop. Mediated"),
CI = ci,
CI_low = c(
stats::quantile(x$d0.sims, probs = 1 - alpha, names = FALSE),
stats::quantile(x$z0.sims, probs = 1 - alpha, names = FALSE),
stats::quantile(x$tau.sims, probs = 1 - alpha, names = FALSE),
stats::quantile(x$n0.sims, probs = 1 - alpha, names = FALSE)
),
CI_high = c(
stats::quantile(x$d0.sims, probs = alpha, names = FALSE),
stats::quantile(x$z0.sims, probs = alpha, names = FALSE),
stats::quantile(x$tau.sims, probs = alpha, names = FALSE),
stats::quantile(x$n0.sims, probs = alpha, names = FALSE)
),
stringsAsFactors = FALSE
)
} else {
out <- data.frame(
Parameter = c(
"ACME (control)", "ACME (treated)", "ADE (control)",
"ADE (treated)", "Total Effect", "Prop. Mediated (control)",
"Prop. Mediated (treated)", "ACME (average)", "ADE (average)",
"Prop. Mediated (average)"
),
CI = ci,
CI_low = c(
stats::quantile(x$d0.sims, probs = 1 - alpha, names = FALSE),
stats::quantile(x$d1.sims, probs = 1 - alpha, names = FALSE),
stats::quantile(x$z0.sims, probs = 1 - alpha, names = FALSE),
stats::quantile(x$z1.sims, probs = 1 - alpha, names = FALSE),
stats::quantile(x$tau.sims, probs = 1 - alpha, names = FALSE),
stats::quantile(x$n0.sims, probs = 1 - alpha, names = FALSE),
stats::quantile(x$n1.sims, probs = 1 - alpha, names = FALSE),
stats::quantile(x$d.avg.sims, probs = 1 - alpha, names = FALSE),
stats::quantile(x$z.avg.sims, probs = 1 - alpha, names = FALSE),
stats::quantile(x$n.avg.sims, probs = 1 - alpha, names = FALSE)
),
CI_high = c(
stats::quantile(x$d0.sims, probs = alpha, names = FALSE),
stats::quantile(x$d1.sims, probs = alpha, names = FALSE),
stats::quantile(x$z0.sims, probs = alpha, names = FALSE),
stats::quantile(x$z1.sims, probs = alpha, names = FALSE),
stats::quantile(x$tau.sims, probs = alpha, names = FALSE),
stats::quantile(x$n0.sims, probs = alpha, names = FALSE),
stats::quantile(x$n1.sims, probs = alpha, names = FALSE),
stats::quantile(x$d.avg.sims, probs = alpha, names = FALSE),
stats::quantile(x$z.avg.sims, probs = alpha, names = FALSE),
stats::quantile(x$n.avg.sims, probs = alpha, names = FALSE)
),
stringsAsFactors = FALSE
)
}
out
}
#' @export
standard_error.mediate <- function(model, ...) {
NULL
}
#' @export
p_value.mediate <- function(model, ...) {
info <- insight::model_info(model$model.y, verbose = FALSE)
if (info$is_linear && !model$INT) {
out <- data.frame(
Parameter = c("ACME", "ADE", "Total Effect", "Prop. Mediated"),
p = c(model$d0.p, model$z0.p, model$tau.p, model$n0.p),
stringsAsFactors = FALSE
)
} else {
out <- data.frame(
Parameter = c(
"ACME (control)", "ACME (treated)", "ADE (control)", "ADE (treated)",
"Total Effect", "Prop. Mediated (control)", "Prop. Mediated (treated)",
"ACME (average)", "ADE (average)", "Prop. Mediated (average)"
),
p = c(
model$d0.p, model$d1.p, model$z0.p, model$z1.p, model$tau.p, model$n0.p,
model$n1.p, model$d.avg.p, model$z.avg.p, model$n.avg.p
),
stringsAsFactors = FALSE
)
}
out
}
#' @export
format_parameters.mediate <- function(model, ...) {
params <- insight::find_parameters(model, flatten = TRUE)
params <- insight::trim_ws(gsub("(.*)\\((.*)\\)$", "\\1", params))
names(params) <- params
params[params == "ACME"] <- "Indirect Effect (ACME)"
params[params == "ADE"] <- "Direct Effect (ADE)"
params
}
|
