R/getAssociationSets.R

In mi-erasmusmc/AssociationRuleMining: What the Package Does (One Line, Title Case)

getAssociationSets <- function(algorithm, inputFile, outputFile, minsup, minconf, showID = FALSE) {
  
  spmf.dir <- rJava::.jclassPath()[stringr::str_ends(rJava::.jclassPath(), "spmf.jar")]
  associationAlgorithms <- c("Apriori", "Eclat", "FP-Growth", "Relim", "AllRules", "NonRedRules")
  `%notin%` <- Negate(`%in%`)
  outputID = paste(tolower(showID))
  #maxLengthvalue = jdx::convertToJava(maxLength, scalars.as.objects = TRUE)
  #maxGapvalue = jdx::convertToJava(maxGap, scalars.as.objects = TRUE)
  maxLengthvalue = 1000
  maxGapvalue = 1000
  
  if(algorithm %notin% associationAlgorithms){
    stop("Algorithm is not supported at the moment!")
  }
  
  #Below I replaced the Inf values for maxLength and maxGap with 1000 since Inf is not a Java object for SPAM and prefixSpan
  if (algorithm == "Apriori" ) {
    executable <- paste("java -jar", spmf.dir, "run", algorithm, inputFile, outputFile, minsup, sep = " ") # Removed outputID but should look up for other parameters since some are applicable
  } else {
    if (algorithm == "Eclat") {
      executable <- paste("java -jar", spmf.dir, "run", algorithm, inputFile, outputFile, minsup, outputID, sep = " ")
    } else {
      if (algorithm == "FP-Growth"){
        executable <- paste("java -jar", spmf.dir, "run", "FPGrowth_itemsets", inputFile, outputFile, minsup, sep = " ")
      } else {
       if (algorithm == "Relim"){
         executable <- paste("java -jar", spmf.dir, "run", "Relim", inputFile, outputFile, minsup, sep = " ")
       } else{
         if (algorithm == "AllRules"){
           executable <- paste("java -jar", spmf.dir, "run", "FPGrowth_association_rules", inputFile, outputFile, minsup, minconf, sep = " ")
         } else {
           if (algorithm == "NonRedRules"){
             executable <- paste("java -jar", spmf.dir, "run", "MNR", inputFile, outputFile, minsup, minconf, sep = " ")
           }
         }
       } 
      } 
    }
  }
  cat(paste("The command line that has been running is:", print(executable)))
  system(executable)
}