R中使用glmnet绘图时的图例标签错误。

Question

我修改了这篇帖子中的函数（在R中为glmnet图形添加标签），并按如下方式添加了图例：

library(glmnet)
fit = glmnet(as.matrix(mtcars[-1]), mtcars[,1])

lbs_fun <- function(fit, ...) {
        L <- length(fit$lambda)
        x <- log(fit$lambda[L])
        y <- fit$beta[, L]
        labs <- names(y)
        text(x, y, labels=labs, ...)
        legend('topright', legend=labs, col=1:length(labs), lty=1) # <<< ADDED BY ME
}
plot(fit, xvar="lambda")
lbs_fun(fit)

然而，我的图表上的文本标签和图例中的文本标签不匹配。变量“am”的颜色明显错误。错误在哪里？谢谢您的帮助。

- rnso

Answer 1

plot(fit, xvar="lambda") 使用函数 matplot。默认情况下，matplot 使用 6 种颜色并进行循环使用。因此，您需要相应地创建图例：

lbs_fun <- function(fit, ...) {
        L <- length(fit$lambda)
        x <- log(fit$lambda[L])
        y <- fit$beta[, L]
        labs <- names(y)
        text(x, y, labels=labs, ...)
        legend('topright', legend=labs, col=1:6, lty=1) # only 6 colors
}

- user3710546

---

我认为当我有超过6个变量时，这会错误地为我标记颜色。 - Hatshepsut

Answer 2

您可以在plot调用中指定颜色以匹配您为图例选择的颜色：

lbs_fun <- function(fit, ...) {
        L <- length(fit$lambda)
        x <- log(fit$lambda[L])
        y <- fit$beta[, L]
        labs <- names(y)
        text(x, y, labels=labs, ...)
        legend('topright', legend=labs, col=1:length(labs), lty=1) # <<< ADDED BY ME
}
plot(fit, xvar="lambda", col=1:dim(coef(fit))[1])
lbs_fun(fit)

- Jota