R/CDMEANMM.R

In STPGA: Selection of Training Populations by Genetic Algorithm

CDMEANMM <-
  function(Train,Test, Kinv, K, lambda=1e-5, C=NULL, Vg=NULL, Ve=NULL){
    ntrain<-length(Train)
    M<-matrix(-1/ntrain,ntrain,ntrain)+diag(ntrain)
    namesinPforfactorlevels<-rownames(Kinv)
    factorTrain<-factor(as.character(Train), levels=namesinPforfactorlevels)
    factorTest<-factor(as.character(Test), levels=namesinPforfactorlevels)
    Ztrain<-as.matrix(model.matrix(~factorTrain-1))
    Ztest<-as.matrix(model.matrix(~factorTest-1))
    if (!is.null(C)){Ztest<-C%*%Ztest}
    CDMAT0<-Ztest%*%(K-lambda*solve(crossprod(Ztrain,M%*%Ztrain)+lambda*Kinv))%*%t(Ztest)
    CDMAT1<-Ztest%*%(K)%*%t(Ztest)
    return(-mean(diag(CDMAT0)/diag(CDMAT1)))
  }