R/parse-fit_lm.R

In kota7/outreg: Regression Table for Publication

make_coef_part.lm <- function(fit, modelname, robust = FALSE, small = TRUE, ...)
{
  out <- data.frame(modelname = modelname,
                    variable = names(coefficients(fit)),
                    coef = coefficients(fit),
                    stringsAsFactors = FALSE)

  se <- rep(NA, nrow(out))
  vcov_func <- ifelse(robust, sandwich, vcov)
  se[!is.na(coefficients(fit))] <- sqrt(diag(vcov_func(fit)))
  out$se <- se

  out$tv <- out$coef / out$se

  if (small) {
    out$pv <- pt(-abs(out$tv), df.residual(fit))*2
  } else {
    out$pv <- pnorm(-abs(out$tv))*2
  }
  rownames(out) <- NULL
  out
}


make_stat_part.lm <- function(fit, modelname, ...)
{
  data.frame(modelname = modelname,
             variable = '',
             nobs = length(fitted.values(fit)),
             R2 = summary(fit)$r.squared,
             adjR2 = summary(fit)$adj.r.squared,
             aic = AIC(fit),
             stringsAsFactors = FALSE)
}