Logistic回归训练集的功能

Question

I am trying to create a function to test a logistic regression model developed on a training set.

For example

train <- filter(y, folds != i)
test <- filter(y, folds == i)

I want to be able to use the formula for different data sets. For example, if I were to take y to be a response variable such as “ low ” in the birthwt data set and x to be the explanatory variables eg “age", “race” how would I implement these arguments into glm.train formula without having to type the function separately for different data sets ?

glm.train <- glm(y ~x, family = binomial, data =  train)

Answer 1

You can use reformulate to create a formula based on strings:

x <- c("age", "race")
y <- "low"

form <- reformulate(x, response = y)
# low ~ age + race

Use this formula for glm :

glm.train <- glm(form, family = binomial, data =  train)