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
|
#' @include get_coef.R
#' @rdname get_coef
#' @export
get_coef.glmgee <- function(model, ...) {
b <- model$coefficients
b <- stats::setNames(as.vector(b), row.names(b))
return(b)
}
#' @include set_coef.R
#' @rdname set_coef
#' @export
set_coef.glmgee <- function(model, coefs, ...) {
out <- model
idx <- match(row.names(out$coefficients), names(coefs))
out$coefficients[, 1] <- coefs[idx]
return(out)
}
#' @include get_predict.R
#' @rdname get_predict
#' @export
get_predict.glmgee <- function(model, newdata, ...) {
Yhat <- stats::predict(model, newdata = newdata, type = "response")
out <- data.table(estimate = as.vector(Yhat))
out <- add_rowid(out, newdata)
return(out)
}
|