这是我使用sourceCpp来源的CPP文件:

#include <RcppArmadillo.h>
// [[Rcpp::depends(RcppArmadillo)]]

using namespace Rcpp;

// [[Rcpp::export]]
List mylm(NumericVector yr, NumericMatrix Xr) {

    int n = Xr.nrow(), k = Xr.ncol();

    arma::mat X(Xr.begin(), n, k, false);       // reuses memory and avoids extra copy
    arma::colvec y(yr.begin(), yr.size(), false);

    arma::colvec coef = arma::solve(X, y);      // fit model y ~ X
    arma::colvec resid = y - X*coef;            // residuals

    double sig2 = arma::as_scalar( arma::trans(resid)*resid/(n-k) );
                                                // std.error of estimate
    arma::colvec stderrest = arma::sqrt( sig2 * arma::diagvec( arma::inv(arma::trans(X)*X)) );

    return Rcpp::List::create(
        Rcpp::Named("coefficients") = coef,
        Rcpp::Named("stderr")       = stderrest
    ) ;

}
.

(代码从@romainfrançois答案到这个问题:使用`sourcecpp`编译`fastlm`

然后在R:

sourceCpp('b.cpp')
set.seed(1)
x = matrix(rnorm(100), 25, 4)
y = rnorm(25)
mylm(y, x)
summary(lm(y~x))

mylm(y, x)
# $coefficients
#           [,1]
# [1,]  0.068978
# [2,]  0.117632
# [3,] -0.029917
# [4,] -0.168648
# 
# $stderr
#         [,1]
# [1,] 0.17970
# [2,] 0.24312
# [3,] 0.15138
# [4,] 0.21266

summary(lm(y~x))

# Call:
# lm(formula = y ~ x)
# 
# Residuals:
#    Min     1Q Median     3Q    Max 
# -0.869 -0.487 -0.271  0.410  1.831 
# 
# Coefficients:
#             Estimate Std. Error t value Pr(>|t|)
# (Intercept)   0.1122     0.1718    0.65     0.52
# x1            0.0499     0.1845    0.27     0.79
# x2            0.1076     0.2470    0.44     0.67
# x3           -0.0435     0.1549   -0.28     0.78
# x4           -0.1750     0.2158   -0.81     0.43
# 
# Residual standard error: 0.835 on 20 degrees of freedom
# Multiple R-squared:  0.054,     Adjusted R-squared:  -0.135 
# F-statistic: 0.285 on 4 and 20 DF,  p-value: 0.884
.

有帮助吗?

解决方案

默认情况下,即使您传递的设计矩阵不包括初始列,R的LM也适合带有截距的型号。所以你会看到以下内容是相同的:

lm(y ~ x - 1)
mylm(y, x)
.

如果您想要'常规'模型,您需要修改设计矩阵以具有所有1s的第一列:

lm(y ~ x)
mylm(y, cbind(1, x))
.

将提供相同的结果。

许可以下: CC-BY-SA归因
不隶属于 StackOverflow
scroll top