R/log_reg_models.R

In graggsd/sgcleanup: Useful Functions to Clean and Reshape Data

#' Adds an odds Ratio and 95% confidence interval
#'
#' @export
add_OR_bounds <- function(x, ...) {
    UseMethod("add_OR_bounds", x)
}

add_OR_bounds.data.frame <- function(x, ...) {
    stopifnot("estimate" %in% colnames(x),
              "std.error" %in% colnames(x))
    x$OR <- exp(x$estimate)
    x$Lower <- exp(x$estimate - 1.96*x$std.error)
    x$Upper <- exp(x$estimate + 1.96*x$std.error)
    return(x)
}

add_OR_bounds.tbl_df <- function(x, ...) {
    stopifnot("estimate" %in% colnames(x),
              "std.error" %in% colnames(x))
    x$OR <- exp(x$estimate)
    x$Lower <- exp(x$estimate - 1.96*x$std.error)
    x$Upper <- exp(x$estimate + 1.96*x$std.error)
    return(x)
}