regression_report.R en R

Ajustar una regresión múltiple e imprimir una tabla prolija de coeficientes.

#!/usr/bin/env Rscript
# Ajusta un modelo lineal sobre mtcars e imprime un resumen ordenado.

fit_model <- function(data) {
  lm(mpg ~ wt + hp + factor(cyl), data = data)
}

print_coefficients <- function(model) {
  coefs <- summary(model)$coefficients
  cat(sprintf("%-18s %9s %9s %9s\n",
              "term", "estimate", "std.error", "p.value"))
  for (term in rownames(coefs)) {
    cat(sprintf("%-18s %9.3f %9.3f %9.4f\n",
                term, coefs[term, 1], coefs[term, 2], coefs[term, 4]))
  }
}

model <- fit_model(mtcars)
print_coefficients(model)

cat("\nr.squared:    ", round(summary(model)$r.squared, 4), "\n")
cat("adj.r.squared:", round(summary(model)$adj.r.squared, 4), "\n")
cat("rmse:         ", round(sqrt(mean(resid(model)^2)), 4), "\n")
cat("aic:          ", round(AIC(model), 2), "\n")

Cómo funciona

La fórmula mezcla predictores numéricos con un factor.
summary(model)$coefficients alimenta una tabla formateada.
R cuadrado, RMSE y AIC resumen el ajuste.

Palabras clave y builtins usados aquí

for
function
in

El intento, en números

Líneas: 24
Caracteres a escribir: 765
Tokens: 199
Ritmo de tres estrellas: 105 tpm

Al ritmo de tres estrellas de 105 tokens por minuto, este intento toma unos 114 segundos.

Escribe este fragmento

Paso 1 de 2 en Bis; paso 10 de 11 en Estadística.

← Anterior Siguiente →