library(ISLR2)
library(glmnet)
data(Hitters)
Hitters <- na.omit(Hitters)
X <- model.matrix(Salary ~ ., data=Hitters)[, -1]
y <- Hitters$Salary
set.seed(42)
cv_lasso <- cv.glmnet(X, y, alpha=1, nfolds=10)
cat("lambda.min:", round(cv_lasso$lambda.min, 4), "\n")lambda.min: 2.2203
cat("lambda.1se:", round(cv_lasso$lambda.1se, 4), "\n")lambda.1se: 83.5934
cat("\nNumber of non-zero coefficients:\n")
Number of non-zero coefficients:
cat(" at lambda.min:", sum(coef(cv_lasso, s="lambda.min") != 0) - 1, "\n") at lambda.min: 14
cat(" at lambda.1se:", sum(coef(cv_lasso, s="lambda.1se") != 0) - 1, "\n") at lambda.1se: 5
plot(cv_lasso)