R/run_EGA_combinations.R

In InterconectaR: Psychological Network Analysis and Psychometric Insights

#' @title Run EGA Combinations
#' @description Runs EGA with multiple combinations of correlation methods and algorithms.
#' @param data Data frame with the data.
#' @param corr Character vector of correlation methods.
#' @param algorithm Character vector of community detection algorithms.
#' @param leiden_args List of additional arguments for the Leiden algorithm.
#' @return A list of EGA results with class "EGA_combinations".
#' @export
#' @examples
#' set.seed(123)
#' n <- 300
#' f1 <- rnorm(n)
#' f2 <- 0.3 * f1 + rnorm(n)
#' items <- data.frame(
#'   sapply(1:4, function(i) 0.7 * f1 + rnorm(n, 0, 0.7)),
#'   sapply(1:4, function(i) 0.7 * f2 + rnorm(n, 0, 0.7))
#' )
#' names(items) <- c(paste0("A", 1:4), paste0("B", 1:4))
#'
#' combos <- run_EGA_combinations(items, corr = c("pearson", "spearman"),
#'                                algorithm = c("louvain", "walktrap"))
#' names(combos)
#' @importFrom EGAnet EGA
run_EGA_combinations <- function(data,
                                 corr = c("cor_auto", "pearson", "spearman"),
                                 algorithm = c("leiden", "louvain", "walktrap"),
                                 leiden_args = list(objective_function = "CPM",
                                                    resolution_parameter = 1.0)) {

  # Validar parametros de entrada
  corr <- match.arg(corr, several.ok = TRUE)
  algorithm <- match.arg(algorithm, several.ok = TRUE)

  # Crear combinaciones posibles
  combinations <- expand.grid(corr = corr, algorithm = algorithm, stringsAsFactors = FALSE)

  # Lista para resultados
  results <- list()

  # Iterar sobre combinaciones
  for(i in seq_len(nrow(combinations))) {
    current_corr <- combinations$corr[i]
    current_alg <- combinations$algorithm[i]

    # Configurar parametros especificos para Leiden
    if(current_alg == "leiden") {
      args <- list(
        data = data,
        corr = current_corr,
        model = "glasso",
        algorithm = current_alg,
        plot.EGA = FALSE
      )
      args <- modifyList(args, leiden_args)
    } else {
      args <- list(
        data = data,
        corr = current_corr,
        model = "glasso",
        algorithm = current_alg,
        plot.EGA = FALSE
      )
    }

    # Nombre unico para la combinacion
    combo_name <- paste(current_corr, current_alg, sep = ".")

    # Ejecutar con manejo de errores y advertencias
    results[[combo_name]] <- tryCatch({
      suppressWarnings(do.call(EGA, args))
    }, error = function(e) {
      message("Error en ", combo_name, ": ", e$message)
      return(NULL)
    })
  }

  # Filtrar resultados nulos
  results <- results[!sapply(results, is.null)]

  # Agregar metadatos
  attr(results, "combinations") <- combinations
  attr(results, "leiden_args") <- leiden_args
  class(results) <- c("EGA_combinations", class(results))

  return(results)
}