Rcpp несколько функций в файле и без функции соответствия - PullRequest
Новая
       7

Rcpp несколько функций в файле и без функции соответствия

0 голосов
/ 28 апреля 2018

Я пытаюсь запустить функцию weights ниже в R. Функции mN и PsiN ниже работают индивидуально, и мне не нужно экспортировать их в R (т. Е. Их единственная цель - сохранить веса функций выглядит аккуратнее).

По какой-то причине только mN выдает мне ошибку "нет соответствующего вызова функции" внутри функции weights. Есть идеи, почему? 

#include <RcppArmadillo.h>
// [[Rcpp::depends(RcppArmadillo)]]
using namespace Rcpp;
using namespace arma;

arma::vec mN(double kappa, double ni, arma::vec m, arma::vec C) {
  arma::vec mN; 
  double kappaN;
  kappaN = kappa + ni;
  mN = (kappa/kappaN) * m + (ni/kappaN) * C;
  return mN; 
}
arma::mat PsiN(double ni, double kappa, double kappaN,
               arma::mat y, arma::vec indic, arma::vec C, arma::vec m,
               arma::mat Psi) { 
  double r = indic.n_elem;
  double p = y.n_cols; 
  double n = y.n_rows;
  arma::mat S(p, p); 
  arma::mat PN(p, p); 
  if (r == 1) {
    PN = Psi + kappa*ni/kappaN * ( (C - m ) * (C - m).t() ); 
  } else {
    for (int i = 0; i < n; i++) {
      S += y.row(i).t() * y.row(i);
    }
    PN = Psi + S + kappa*ni/kappaN * ( (C - m ) * (C - m).t() ) ;  
  }
  return PN;
}

// [[Rcpp::export]]

arma::vec weights(int alpha, int v, int kappa, int m,
                  arma::vec nj, arma::mat x, arma::mat Psi, 
               List C, List indic) {
  int p = x.n_cols;
  int n = x.n_rows;
  int kappaN;
  int vN; 
  int crp; 
  arma::vec Q; 
  arma::vec mu;
  arma::mat Sigma;

  for (int i = 0; i < n; i++) {
    kappaN = kappa + nj[i];
    vN     = v + nj[i] - p + 1;
    Sigma  = PsiN(nj[i], kappa, kappaN, x, indic[i], C[i], m, Psi);
    Sigma  = Sigma*(kappaN + 1)/(kappaN*(vN - p + 1));
    mu     = mN(kappa, kappaN, nj[i], m, C[i]); 
    crp    = log(nj[i]) - log(n + alpha - 1);
    Q[i]   = (crp - lgamma((vN + p)/2) - lgamma(vN/2) - p/2*(log(vN) + log(
      datum::pi)) - 1/2*log_det(Sigma) - (vN + p)/2 * (log(1 + 1/v*(x - mu).t()*Sigma.i()*(x-mu)))) ;
  }

 return Q;
}

Ответы [ 2 ]

0 голосов
/ 28 апреля 2018

Ваша подпись функции для mN равна 

arma::vec mN(double, double, arma::vec, arma::vec)

Но вы называете это 

mu     = mN(kappa, kappaN, nj[i], m, C[i]);

, где kappa - int, kappaN - int, nj[i] - double, m - int и C - List из SEXP с.

Обратите внимание, что, в отличие от R, вы вообще не можете смешивать двойные и целые числа в C ++. В частности, я также укажу, что объявления типа 

double r = indic.n_elem;
double p = y.n_cols; 
double n = y.n_rows;

должно быть int с или unsigned int с.

0 голосов
/ 28 апреля 2018

Сигнатуры функций в определении: 

arma::vec mN(double kappa, double ni, arma::vec m, arma::vec C)

не соответствует его использованию: 

mu     = mN(kappa, kappaN, nj[i], m, C[i]);

Вы должны решить, нужно ли функции 4 или 5 аргументов.

...