Exemplo n.º 1
0
  public static void main(String[] args) throws Exception {
    Neo4jHandler neo4jHandler = Neo4jHandler.getInstance("/media/disk/master/Noe4j/reuters");
    ReutersDataset datasetHandler =
        new ReutersDataset("/media/disk/master/Noe4j/datasets/reuters_mod");

    long startTime = System.currentTimeMillis();

    SinglePass singlePassAlgorithm = new SinglePass();
    double threshold = 0.1;
    Hashtable<String, Neo4jCluster> clusters =
        singlePassAlgorithm.perform(datasetHandler, neo4jHandler, threshold, 5);
    long endTime = System.currentTimeMillis();

    FMeasure fmeasureCalculate = new FMeasure();
    fmeasureCalculate.calculate(clusters, datasetHandler, neo4jHandler);
    System.out.println("*********************");
    System.out.println("Total elapsed time in execution  is :" + (endTime - startTime));

    System.out.println("******* For Threshold = " + threshold);
    System.out.println("Fmeasure = " + fmeasureCalculate.getFmeasure());
    System.out.println("Precision = " + fmeasureCalculate.getPrecision());
    System.out.println("Recall = " + fmeasureCalculate.getRecall());
    System.out.println("*********************");

    //		System.out.println("Number of documents = "+ datasetHandler.numberOfDocuments());
    //		Enumeration e = clusters.keys();
    //		while (e.hasMoreElements()) {
    //			String clusterID = (String) e.nextElement();
    //			System.out.println("Cluster = " + clusterID + " has number of documents = " +
    // clusters.get(clusterID).getDocumentIDs().size());
    //			ArrayList<Neo4jDocument> documents =
    // clusters.get(clusterID).getDocumentsList(datasetHandler, neo4jHandler);
    //			for (int i = 0; i < documents.size(); i++) {
    //
    //	System.out.println(datasetHandler.getDocument(documents.get(i).getDocumentID()).getOrginalCluster());
    //			}
    //
    //	System.out.println(("*********************************************************************************"));
    //		}
    neo4jHandler.registerShutdownHook();
  }