R/PEVMEAN2.R

In STPGA: Selection of Training Populations by Genetic Algorithm

#correct version but very close to the previous one. 
PEVMEAN2 <-
  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}
    PEV<-PTest%*%solve(crossprod(PTrain)+lambda*diag(ncol(PTrain)),t(PTrain))
    PEVmean<-mean(diag(tcrossprod(PEV)))
    return(PEVmean)
  }