package ca.pfv.spmf.test;
import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.net.URL;
import ca.pfv.spmf.algorithms.associationrules.agrawal94_association_rules.AlgoAgrawalFaster94;
import ca.pfv.spmf.algorithms.frequentpatterns.cfpgrowth.AlgoCFPGrowth;
import ca.pfv.spmf.patterns.itemset_array_integers_with_count.Itemsets;
/**
* Example of how to mine all association rules with CFPGROWTH
* and use the lift, and save the result to a file,
* from the source code.
*
* @author Philippe Fournier-Viger (Copyright 2013)
*/
public class MainTestAllAssociationRules_CFPGrowth_saveToFile_withLift {
public static void main(String [] arg) throws IOException{
String input = fileToPath("contextIGB.txt");
String output = ".//output.txt";
String MISfile = fileToPath("MIS.txt");
// STEP 1: Applying the CFP-GROWTH algorithm to find frequent itemsets
AlgoCFPGrowth cfpgrowth = new AlgoCFPGrowth();
Itemsets patterns = cfpgrowth.runAlgorithm(input, null, MISfile);
// patterns.printItemsets(database.size());
int databaseSize = cfpgrowth.getDatabaseSize();
cfpgrowth.printStats();
// STEP 2: Generating all rules from the set of frequent itemsets (based on Agrawal & Srikant, 94)
double minlift = 0.1;
double minconf = 0.50;
AlgoAgrawalFaster94 algoAgrawal = new AlgoAgrawalFaster94();
algoAgrawal.runAlgorithm(patterns, output, databaseSize, minconf, minlift);
algoAgrawal.printStats();
}
public static String fileToPath(String filename) throws UnsupportedEncodingException{
URL url = MainTestAllAssociationRules_CFPGrowth_saveToFile_withLift.class.getResource(filename);
return java.net.URLDecoder.decode(url.getPath(),"UTF-8");
}
}