R: Автоматизация экстракции уравнения линейной регрессии от lm

Question

Кто-нибудь знает о существующей функции для извлечения полного линейного уравнения из объекта lm?

Пусть у меня есть:

lm1 = lm(y~x1+x2...xn, data=df) 

Для этого курса в регрессии я везу, профессор неоднократно хочет полученное уравнение регрессии в виде: е (у) = b1 + b2x1 [...] BNX (п-1).

В настоящее время, я делаю что-то вроде этого:

(paste("y=", coef(lm1)[1], '+', coef(lm1[2]), '*x2', [...]) 

Он пронес в течение многих недель, как это. Не было бы огромной проблемой копировать и вставлять функцию вставки выше, но он хочет, чтобы вместо нее отображались фактические метки переменных вместо y, x1 и т. Д. Как вы можете видеть, моя рука и ум болят от многократного выполнения этого.

Я преодолел свой порог боли для этого, и сегодня начал думать о создании моей собственной функции для этого, но я просто проверяю, знает ли кто-нибудь о существующей функции, которая делает это автоматически.

Не думаю, что создать функцию, которая выполняет одну и ту же функцию вставки, кроме переменной длины для числа коэффициентов, будет чрезвычайно сложно, но это просто то, что я предпочел бы пойти с существующим решением, учитывая конечный продукт - такая бесполезная конструкция.

Примечание: это очень похоже, но не совсем тот же вопрос, поставленный здесь: Extract Formula From lm with Coefficients (R)

Почему он отличается? Эти вопросы касаются одного случая: «Как вы извлекаете уравнение линейной регрессии из объекта lm?» Этот вопрос заключается в том, что «существует ли существующий базовый метод или (если нет) метод для систематически, получающий уравнение линейной регрессии от объекта lm. Это особенно заметно, когда вы видите ответ, поставленный здесь, против принятого ответа на другая страница

Answer 1

Мой наивный подход к этой проблеме также должен был бы настроить свою собственную функцию, используя paste0():..

regEq <- function(lmObj, dig) { 
    paste0("y = ", 
     paste0(
      c(round(lmObj$coef[1], dig), round(sign(lmObj$coef[-1])*lmObj$coef[-1], dig)), 
      c("", rep("*", length(lmObj$coef)-1)), 
      paste0(c("", names(lmObj$coef)[-1]), c(ifelse(sign(lmObj$coef)[-1]==1," + "," - "), "")), 
      collapse="" 
     ) 
    ) 
} 

, где он получает немного грязный из-за признаков и перехватывать Давайте посмотрим на простой вызов :

> fit <- lm(-mpg ~ cyl + hp + drat, data=mtcars) 
> fit 

Call: 
lm(formula = -mpg ~ cyl + hp + drat, data = mtcars) 

Coefficients: 
(Intercept)   cyl   hp   drat 
    -22.51406  1.36060  0.02878  -2.84090 

> regEq(fit,3) 
[1] "y = -22.514 + 1.361*cyl + 0.029*hp - 2.841*drat" 

EDIT: по запросу. комментарий: Для того, чтобы заменить y с именем переменной и изменить оператор взаимодействия с * переписывания:

regEq <- function(lmObj, dig) { 
    gsub(":", "*", 
     paste0(
      names(lmObj$model)[1]," = ", 
      paste0(
       c(round(lmObj$coef[1], dig), round(sign(lmObj$coef[-1])*lmObj$coef[-1], dig)), 
       c("", rep("*", length(lmObj$coef)-1)), 
       paste0(c("", names(lmObj$coef)[-1]), c(ifelse(sign(lmObj$coef)[-1]==1," + "," - "), "")), 
       collapse="" 
      ) 
     ) 
    ) 
} 

> regEq(lm(mpg ~ hp * drat, data=mtcars), 3) 
[1] "mpg = 5.55 - 0.013*hp + 6.069*drat - 0.01*hp*drat"