Skip to content
Snippets Groups Projects
predicteurs.R 416 B
Newer Older
Heloise Chevalier's avatar
Heloise Chevalier committed
regresseur <- function(dataset) {
  # Chargement de l'environnement
  load("env.Rdata")
  cv.out <- cv.glmnet(as.matrix(reg.train.X),reg.train.y,alpha=1)
  #model.reg <- glmnet(as.matrix(reg.train.X),reg.train.y,lambda=cv.out$lambda.min,alpha=1)
Heloise Chevalier's avatar
Heloise Chevalier committed
  model.reg <- lm(reg.train.y ~ ., data = reg.train.X)
  data.test <- dataset[,reg.mask]
  predictions <- predict(model.reg, newdata = data.test)
  return(predictions)
}