R/CDMEAN2.R

In STPGA: Selection of Training Populations by Genetic Algorithm

CDMEAN2 <-
  function(Train,Test, P, lambda=1e-5, C=NULL){
    PTrain<-P[rownames(P)%in%Train,]
    PTest<-P[rownames(P)%in%Test,]
    if (length(Test)==1){PTest=matrix(PTest, nrow=1)}
    if (!is.null(C)){PTest<-C%*%PTest}
    CD<-PTest%*%solve(crossprod(PTrain)+lambda*diag(ncol(PTrain)),t(PTrain))
    CDmean<-mean(diag(tcrossprod(CD))/diag(tcrossprod(PTest)))
    return(CDmean)
  }