package ca.pfv.spmf.test;
import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.net.URL;
import ca.pfv.spmf.algorithms.associationrules.MNRRules.AlgoMNRRules;
import ca.pfv.spmf.algorithms.frequentpatterns.zart.AlgoZart;
import ca.pfv.spmf.algorithms.frequentpatterns.zart.TZTableClosed;
import ca.pfv.spmf.input.transaction_database_list_integers.TransactionDatabase;
/**
* Example of how to generate minimal non redundant
* association rules in source code.
*
* @author Philippe Fournier-Viger, 2008
*/
public class MainTestMNRRules_saveToFile {
public static void main(String[] args) throws IOException {
System.out.println("STEP 1 : EXECUTING THE ZART ALGORITHM TO FIND CLOSED ITEMSETS AND MINIMUM GENERATORS");
String input = fileToPath("contextZart.txt");
String output = ".//output.txt";
//
double minsup = 0.6; // minimum support
double minconf = 0.6; // minimum confidence
TransactionDatabase database = new TransactionDatabase();
try {
database.loadFile(input);
} catch (Exception e) {
e.printStackTrace();
}
database.printDatabase();
// Applying the Zart algorithm
AlgoZart zart = new AlgoZart();
TZTableClosed results = zart.runAlgorithm(database, minsup);
zart.printStatistics();
// // PRINT RESULTS FROM THE ZART ALGORITHM
// int countClosed=0;
// int countGenerators=0;
// System.out.println("===================");
// for(int i=0; i< results.levels.size(); i++){
// System.out.println("LEVEL : " + i);
// for(Itemset closed : results.levels.get(i)){
// System.out.println(" CLOSED : " + closed.toString() + " supp : " + closed.getAbsoluteSupport());
// countClosed++;
// System.out.println(" GENERATORS : ");
// for(Itemset generator : results.mapGenerators.get(closed)){
// countGenerators++;
// System.out.println(" =" + generator.toString());
// }
// }
// }
System.out.println("STEP 2 : CALCULATING MNR ASSOCIATION RULES");
// Run the algorithm to generate MNR rules
AlgoMNRRules algoMNR = new AlgoMNRRules();
algoMNR.runAlgorithm(output, minconf, results, database.size());
}
public static String fileToPath(String filename) throws UnsupportedEncodingException{
URL url = MainTestMNRRules_saveToFile.class.getResource(filename);
return java.net.URLDecoder.decode(url.getPath(),"UTF-8");
}
}