Как использовать vars в lm в собственной функции?

Question

Я хочу написать функцию, которая вычисляет линейную регрессию на основе входных данных.

Я могу построить функцию, но когда я ее вызываю (например, myregression(i1,i2), это приведет к ошибке)

myregression <- function(input1, input2) {
   model <- lm(data = trainData, example ~ input1 + input2)
}

Как я могу использовать вход в функции lm?

Answer 1

Внутри функции мы можем использовать paste для создания формулы

myregression <- function(input1, input2) {
    model <- lm(data = trainData, paste0("example ~", input1, " + ", input2))
     }

---

Или другой вариант reformulate

myregression <- function(input1, input2) {
      model <- lm(data = trainData, reformulate(c(input1, input2), "example"))
  }

и вызовите функцию как

myregression("i1", "i2")