分配了矩阵列标准偏差的向量正在更改其值(RcppArmadillo)

时间:2019-01-25 21:05:48

标签: c++ r rcpp armadillo

我正在研究一个RcppArma软件包,并且正在对提升算法的设计矩阵进行居中和标准化,这是简化后的代码:

// [[Rcpp::depends(RcppArmadillo)]]

#include <RcppArmadillo.h>
using namespace Rcpp;
using namespace arma;

// [[Rcpp::export]]
List centering(arma::mat & X) {

   int p = X.n_cols;

   rowvec meanx(p);
   rowvec sigmax(p);

    meanx=mean(X,0);
    sigmax=stddev(X,0);

    for(int j=0;j<p;j++)
    {
      X.col(j)=X.col(j)-meanx(j);
      X.col(j)=X.col(j)/sigmax(j);
    }

  return List::create(Named("sigma") = sigmax, Named("X") = X);
}

对中工作正常,但对中后向量“ sigmax”的所有值均等于“ 1”,因此向量以某种方式将自身更新为对中矩阵X的新标准偏差,而无需重新分配。我需要原始值来对系数进行逆变换。为什么这样做呢?如何避免这种情况?

我用以下代码测试了R中的功能:

set.seed(42)    
X <- replicate(10, rnorm(100, 5, 3))
res <- centering(X)
res <- centering(X)

第二次致电时出现问题。第一次工作。

1 个答案:

答案 0 :(得分:2)

简而言之:请勿在函数定义中的参数&旁使用引用(X)。这会激活RcppArmadillo对an advanced constructor for arma::mat的使用,它会重新使用 R 对象的内存(参见include/RcppArmadilloWrap.h

因此,要解决此问题,请访问:

List centering_reuse_memory(arma::mat & X) {
                                    # ^ reference/reuse
  # Routine given in OP
}

收件人:

List centering_new_memory(arma::mat X) {
                                 # ^ Direct copy

  # Routine given in OP
}

了解共享...

让我们看一下对象如何变化。

# Create the original object
set.seed(42)    
X <- replicate(3, rnorm(5, 5, 3))

# Create a duplicate object not sharing memory with X
set.seed(42)    
X_clone <- replicate(3, rnorm(5, 5, 3))

# View object
X
#         [,1]      [,2]       [,3]
# [1,] 9.112875  4.681626  8.9146090
# [2,] 3.305905  9.534566 11.8599362
# [3,] 6.089385  4.716023  0.8334179
# [4,] 6.898588 11.055271  4.1636337
# [5,] 6.212805  4.811858  4.6000360

# Check equality
all.equal(X, X_clone)
# [1] TRUE

现在,使用参数arma::mat & X传递函数运行

res <- centering_reuse_memory(X)

# Verify results are the same.
all.equal(X, X_clone)
# [1] "Mean relative difference: 8.387859"

# Check X manually to see what changed... 
X
#             [,1]       [,2]       [,3]
# [1,]  1.34167459 -0.7368308  0.6566715
# [2,] -1.45185917  0.8327104  1.3376293
# [3,] -0.11282266 -0.7257062 -1.2116948
# [4,]  0.27645691  1.3245379 -0.4417510
# [5,] -0.05344967 -0.6947113 -0.3408550

为什么有区别?好吧,通过使用引用,可以将 C ++ 函数中的修改传播回驻留在 R 中的X变量,该变量与匹配存储的对象在res$X

# Verify R's X matches the saved C++ routine X modification
all.equal(X, res$X)
# [1] TRUE