---

EM.java

gehe zur Dokumentation dieser Datei

/*
 * $Source: /shared/cvsroot/diplom/app/src/java/de/picana/clusterer/EM.java,v $
 * $Author: mstolpe $
 * $Date: 2003/04/22 09:51:27 $
 * $Revision: 1.6 $
 * $Release$ 
 *
 * Created on 28. Oktober 2002, 17:33
 *
 * Copyright 2002 by Marco Stolpe
 */

package de.picana.clusterer;

import de.picana.control.*;
import de.picana.logging.*;

import weka.core.*;
import java.io.*;


public class EM extends Clusterer {
    
    private weka.clusterers.EM algo;
    private int max_iterations;
    private long time_total;

    
    public EM() {
    }

    public void init(ParameterSet params, Logger logger) {
        
        super.init(params, logger);      
        
        max_iterations = -1;
        try {
            max_iterations = Integer.parseInt((String)params.getParameter("max_iterations"));
        } catch (NumberFormatException nfe) {}
    }
    
    public void buildClusterer(Instances set) throws TaskException {
    
        try {
            statwriter.println("algo_clusterer_name:EM");
            statwriter.println("algo_clusterer_training_set:" + infile);
            
            algo = new weka.clusterers.EM();
            algo.setNumClusters(num_clusters);
            if (max_iterations != -1) {
                algo.setMaxIterations(max_iterations);
                statwriter.println("algo_clusterer_maxiter:" + max_iterations);
            }
            time_total = System.currentTimeMillis();
            algo.buildClusterer(set);
            time_total = System.currentTimeMillis() - time_total;
            statwriter.println("time_clusterer_total:" + time_total);
        } catch (Exception e) {
            throw new TaskException(e.toString());
        }
    }
    
    public void saveModel(String filename) throws TaskException {
    
        try {
            File model = new File(filename);
            FileOutputStream fos = new FileOutputStream(model);
            ObjectOutputStream out = new ObjectOutputStream(fos);
            
            out.writeInt(algo.numberOfClusters());
            out.writeObject(algo);
            out.close();
            
        } catch (Exception e) {
            throw new TaskException(e.toString());
        }
    }
}

---