用R中的“ mtcars”数据集绘制LASSO模型

时间:2018-08-09 15:06:32

标签: r regression glmnet lasso

最近我被告知LASSO。看起来有点复杂,我尝试使用mtcars查找示例。以下代码生成了自变量的图:

library("glmnet")
lasso <- glmnet(as.matrix(mtcars[-1]), mtcars[,1], standardize=TRUE, alpha =1)
plot(lasso)

但是这样做没有显示哪一行是哪个变量。

This对此进行了解释,但我无法弄清楚我需要哪部分代码。

This source较易读,但不提供绘图代码。

有人可以帮我弄清楚哪个变量在哪一行吗?

2 个答案:

答案 0 :(得分:1)

使用plot(lasso, label = TRUE)。然后检查rownames(lasso$beta)。在图表中,第i个变量由数值i标记。为了加快循环速度,您可以制作表格:

setNames(rownames(lasso$beta), 1:nrow(lasso$beta))

您可能没有意识到可以检查?plot.glmnet。或者,您可能已经检查了?plot,但在此未找到任何内容。是的,在某些方法中查找通用函数的文档并不容易。

old versionlatest version的包装插图也涵盖了此标签问题。

答案 1 :(得分:1)

您可以使用以下方法将变量号添加到绘图中:

?1533622895

您可以通过以下方式获取系数编号:

plot(lasso, label=T)