我修改了这篇文章中的函数(在 R 中的 glmnet 图中的曲线上添加标签 https://stackoverflow.com/questions/30560689/adding-labels-on-curves-in-glmnet-plot-in-r) 向图中添加图例,如下所示:
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”的颜色显然不正确。错误在哪里?感谢您的帮助。
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
}
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)