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
|
#' @include get_coef.R
#' @rdname get_coef
#' @export
get_coef.data.frame <- function(model, ...) {
checkmate::assert_data_frame(model)
if (!"estimate" %in% colnames(model)) {
stop_sprintf(
"The model object is a data.frame but doesn't contain the column 'estimate'. Make sure these columns are present"
)
}
out <- model$estimate
if ("term" %in% colnames(model)) {
names(out) <- model$term
} else {
names(out) <- seq_along(out)
}
return(out)
}
#' @include set_coef.R
#' @rdname set_coef
#' @export
set_coef.data.frame <- function(model, coefs, ...) {
model$estimate = coefs
return(model)
}
|
