R/MCAoiv.R

In nicolas-robette/GDAtools: Geometric Data Analysis

MCAoiv <- function(X, Z, excl = NULL, row.w = NULL, ncp = 5) {
  if(is.null(row.w)) row.w <- rep(1, nrow(X))
  if(any(sapply(X, FUN = function(x) !is.factor(x)))) stop("variables in X should all be factors")
  if(any(sapply(Z, FUN = function(x) !is.numeric(x) & !is.integer(x) & !is.factor(x)))) stop("variables in Z should all be factor or numeric")
  if(nrow(X) != nrow(Z)) stop("X and Z should have the same number of rows")
  for(i in 1:ncol(Z)) {
    if(!is.factor(Z[,i])) Z[,i] <- (Z[,i]-weighted.mean(Z[,i],row.w)) / descriptio::weighted.sd(Z[,i],row.w)
  }
  mca0 <- speMCA(X, excl = excl)
  ncomp <- sum(mca0$eig$eigen > 1e-10)
  F <- as.data.frame(speMCA(X, excl = excl, ncp = ncomp)$ind$coord)
  lmoiv <- function(y) {
    df <- data.frame(y = y, Z)
    yhat <- stats::residuals(stats::lm(y ~ . , data = df, weights = row.w))
    return(yhat)
  }
  FHAT <- do.call("cbind.data.frame", lapply(F, lmoiv))
  df <- cbind.data.frame(FHAT, X)
  qualsup <- which(sapply(df, is.factor))
  qualsup <- qualsup[qualsup > ncol(FHAT)]
  if(length(qualsup)==0) qualsup <- NULL
  quantsup <- which(sapply(df, function(x) is.numeric(x) | is.integer(x)))
  quantsup <- quantsup[quantsup > ncol(FHAT)]
  if(length(quantsup)==0) quantsup <- NULL
  res <- FactoMineR::PCA(df, scale.unit = FALSE, ncp = ncp, row.w = row.w, quali.sup = qualsup, quanti.sup = quantsup, graph = FALSE)
  pca <- FactoMineR::PCA(F, scale.unit = FALSE, ncp = ncp, row.w = row.w, graph = FALSE)
  res$ratio <- sum(res$eig[,"eigenvalue"]) / sum(pca$eig[,"eigenvalue"])
  class(res) <- c("PCA", "MCAiv", "list")
  return(res)
}