как получить формулу регрессии, используя коэффициенты, включая категориальные переменные - PullRequest
Новая
       101

как получить формулу регрессии, используя коэффициенты, включая категориальные переменные

0 голосов
/ 24 апреля 2020

Интересно, как я могу получить формульную формулу, когда у меня есть коэффициенты, и они включают взаимодействия и категориальные переменные. Вот пример 

data(iris)
iris$catg <- sample(1:4, nrow(iris), replace = T)
iris$catg <- as.factor(iris$catg)
Inx <- sample(nrow(iris),100)
iris$Species <- factor(iris$Species)

train_data <- iris[Inx, ]
test_data <- iris[-Inx,]

Formula <- "Sepal.Length ~ Sepal.Width + Petal.Length + Species:Petal.Width + 
            Sepal.Width:Petal.Length +  Species +  splines::bs(Petal.Width, df = 2, 
            degree = 2) + catg:Sepal.Width"


ModelMatrix <- predict(caret::dummyVars(Formula, train_data, fullRank = T,  
                                        sep = ""), train_data)
cvglm <- glmnet::cv.glmnet(x = ModelMatrix,
                           y = train_data$Sepal.Length, nfolds = 4, keep = TRUE,  
                           alpha = 1, parallel = F, type.measure = 'mse')

coef(cvglm, s = "lambda.1se")

есть предложения? Спасибо.

...