提高拟合多种模型的效率

时间:2019-10-04 07:51:58

标签: r model regression coefficients

感谢那些帮助我解决了许多模型中有关coefficientsp values的问题的人。现在,我可以从许多模型中获得所有系数,p值和AIC值。

md <- "mpg ~ cyl" 
xlist <- c("disp", "hp", "am")
n <- length(xlist)
comb_lst <- unlist(lapply(1:n, function(x) combn(xlist, x, simplify = F)), recursive = F)
md_lst <- lapply(comb_lst, function(x) paste(md, "+", paste(x, collapse = "+")))

coefs <- unlist(lapply(md_lst, function(x) lm(as.formula(x),data = mtcars)$coe[2]))
pvalues <- unlist(lapply(md_lst, function(x) summary(lm(as.formula(x), data = mtcars))$coe["cyl", 4]))
aic <- unlist(lapply(md_lst, function(x) AIC(lm(as.formula(x), data = mtcars))))

从上面的代码中可以看到,最后3行中的每行将一次独立地适合许多模型。这意味着该代码将适合同一组模型3次。对于大数据和许多变量,这可能很耗时。我的问题是如何一次拟合所有模型,然后再获取系数,p值和AIC值。

3 个答案:

答案 0 :(得分:3)

为方便起见,我使用了我编写的package中的函数。如另一个answer所示,可以通过使用summary来避免这种情况。

我们可以做到:

models <- lapply(md_lst, function(x) do.call(lm, list(formula =x,
                                                      data=mtcars)) )
sapply(models, function(x) { cbind(coef(x),
 manymodelr::extract_model_info(x, "p_value")["cyl"], AIC(x) )})

我不想使用软件包(manymodelr)

sapply(models, function(x) {
  cbind(coef(x),coef(summary(x))[,4]["cyl"],
        AIC(x)
        )})

这给出了矩阵列表,每列分别代表系数,p值和AIC。

结果(被截断)

   [[1]]
                       [,1]       [,2]     [,3]
    (Intercept) 34.66099474 0.03366495 167.1456
    cyl         -1.58727681 0.03366495 167.1456
    disp        -0.02058363 0.03366495 167.1456

    [[2]]
                      [,1]         [,2]     [,3]
    (Intercept) 36.9083305 0.0004803752 169.5618
    cyl         -2.2646936 0.0004803752 169.5618
    hp          -0.0191217 0.0004803752 169.5618

    [[3]]
                     [,1]        [,2]     [,3]
    (Intercept) 34.522443 1.28456e-07 167.2191
    cyl         -2.500958 1.28456e-07 167.2191
    am           2.567035 1.28456e-07 167.2191

    [[4]]
                       [,1]      [,2]     [,3]
    (Intercept) 34.18491917 0.1349044 168.0184
    cyl         -1.22741994 0.1349044 168.0184
    disp        -0.01883809 0.1349044 168.0184
    hp          -0.01467933 0.1349044 168.0184

答案 1 :(得分:2)

您可以这样做:

models <- lapply(md_lst, function(x) lm(as.formula(x), data = mtcars))
coefs <- unlist(lapply(models, function(x) x$coef[2]))
pvalues <- unlist(lapply(models, function(x) summary(x)$coef["cyl", 4]))
aic <- unlist(lapply(models, function(x) AIC(x)))

答案 2 :(得分:2)

这使用了将FUN放入combn()调用中的功能。与其他答案一样,它只进行一次线性回归。

md <- "mpg ~ cyl" 
xlist <- c("disp", "hp", "am")

all_models <- unlist(
                lapply(seq_along(xlist),
                       function (k) {
                         combn(xlist,
                               k,
                               FUN = function (x) {
                                 form <- formula(paste(md, '+', paste(x, collapse = '+')))
                                 eval(bquote(lm(.(form), data = mtcars)   ))
                                 },
                               simplify = F
                         )
                         }
                       )
                ,recursive = F
                )

coefs <- sapply(all_models, function(x) x$coe[2])
pvalues <- sapply(all_models, function(x) summary(x)$coe["cyl", 4])
aic <- sapply(all_models, function(x) AIC(x))
相关问题