knngui.java

在weak环境下的knn算法开发 具体需要的说明都在文件包中

/*
 * KNNgui.java
 *
 * Created on December 17, 2005, 6:27 PM
 */

package cap5638;

import javax.swing.*;
import java.io.*;
import weka.gui.arffviewer.ArffPanel;
import java.awt.event.*;
import weka.core.*;
import weka.classifiers.*;
import weka.core.converters.ArffLoader;



/**
 * A graphical implementation to perform k-nearest-neighbor classification
 * @author  JohnChap
 */
public class KnnGui extends javax.swing.JFrame {
    
    private static String title = "John KNN";
    private File trainingFilePath = new File(".");
    private File testFilePath = new File(".");
    private Instances trainingSet;
    private Instances testSet;
    private int classifierSelect = 0;
    private int testSettingsSelect = 0;
    private int[] leaveOutAttribute;
    private boolean resultsToFile = false;
    private ArffPanel arffPanel1;
    private ArffPanel arffPanel2;
    private int testNumber = 0;
    
    /** Creates new form KNNgui */
    public KnnGui() {
        super(title);
        initComponents();
        initRadioButtons();
        initButtons();
        
        //setting up for first run
        useTestSetRadio.setEnabled(false);
        partialDistanceOptionsButton.setEnabled(false);
        
        writeResultsCheckBox.addActionListener(new ActionListener() {
            public void actionPerformed(ActionEvent event) {
                if (!resultsToFile) {
                    resultsToFile = true;
                }
                else resultsToFile = false;
            }
	});
    }
    
    
    /** finish initializing buttons and binding them to actions
     */
    private void initButtons() {
        trainingLoadButton.addActionListener(new ActionListener() {
            public void actionPerformed(ActionEvent event) {
                loadTrainingFile();
                tabPanel2.setEnabled(true);
            }
	});
        
        testLoadButton.addActionListener(new ActionListener() {
            public void actionPerformed(ActionEvent event) {
                loadTestFile();
                useTestSetRadio.setEnabled(true);
            }
	});
        
        partialDistanceOptionsButton.addActionListener(new ActionListener() {
            public void actionPerformed(ActionEvent event) {
                partialDistanceDialog();
            }
	});
        
        startButton.addActionListener(new ActionListener() {
            public void actionPerformed(ActionEvent event) {
                runKnn();
            }
	});  
    }
    
    
    
    
    /** finishes the Radio button initializations not covered in initComponents
     */
    private void initRadioButtons() {
        
        // classifier selection
        simpleKNNRadio.addActionListener(new ActionListener() {
            public void actionPerformed(ActionEvent event) {
                partialDistanceOptionsButton.setEnabled(false);
                classifierSelect = 0;
            }
        });
         
        partialDistanceRadio.addActionListener(new ActionListener() {
            public void actionPerformed(ActionEvent event) {
                partialDistanceOptionsButton.setEnabled(true);
                classifierSelect = 1;
            }
        });
        
        // test settings
        useTestSetRadio.addActionListener(new ActionListener() {
            public void actionPerformed(ActionEvent event) {
                testSettingsSelect = 0;
                startButton.setEnabled(true);
            }
        });
        
        useTrainingSetRadio.addActionListener(new ActionListener() {
            public void actionPerformed(ActionEvent event) {
                testSettingsSelect = 1;
                startButton.setEnabled(true);
            }
        });
        
        
        // grouping radio buttons
        final ButtonGroup classifierButtonGroup = new ButtonGroup();
        classifierButtonGroup.add(simpleKNNRadio);
        classifierButtonGroup.add(partialDistanceRadio);
        
        final ButtonGroup settingsButtonGroup = new ButtonGroup();
        settingsButtonGroup.add(useTrainingSetRadio);
        settingsButtonGroup.add(useTestSetRadio);
    }
    
   /** displays the patial distance dialog.
    */
    private void partialDistanceDialog() {
        String leaveOut = (String)JOptionPane.showInputDialog(partialDistanceOptionsButton,
                    "Enter the indices of attributes to leave out\n"
                    + "(e.g. 1,2,3...)", "Partial distance Options",
                    JOptionPane.PLAIN_MESSAGE);

        printStatus("Read user input: " + leaveOut);
        try {
            if ((leaveOut != null) && (leaveOut.length() > 0)) {
                String[] strArray = leaveOut.split(",");
                int[] indices = new int[strArray.length];
                for (int i = 0; i < strArray.length; i++) {
                    indices[i] = Integer.parseInt(strArray[i]);
                }
            }
        }
        catch (NumberFormatException nfe) {
            printStatus("Partial distance error: " + nfe.getMessage());
        }
        return;
    }
    
    
    /**
     * Prints a message in the status text area
     * @param message  message to be printed to status text area
     */
    private void printStatus(String message) {
        statusTextArea.append(message + "\n");
    }
    
    /** Presents the training set file chooser dialog
     */
    private void loadTrainingFile() {	
        try {
            JFileChooser openChooser = new JFileChooser(trainingFilePath);
            openChooser.showOpenDialog(null);
            trainingFilePath = openChooser.getSelectedFile();
            ArffLoader arffLoad = new ArffLoader();
            arffLoad.setSource(trainingFilePath);
            arffPanel1 = new ArffPanel(trainingFilePath.getPath());
            trainingArffScrollPane.setViewportView(arffPanel1);
            trainingSet = arffLoad.getDataSet();
            printStatus("Training file: " + trainingFilePath.getPath() + " Loaded.");
        }
        catch (Exception e) {
            printStatus("Error loading training set: " + e.getMessage());
        }
  
    } 
    
      /** Presents the test set file chooser dialog
     */
    private void loadTestFile() {
        try {
            JFileChooser openChooser = new JFileChooser(testFilePath);
            openChooser.showOpenDialog(null);
            testFilePath = openChooser.getSelectedFile();
            ArffLoader arffLoad = new ArffLoader();
            arffLoad.setSource(trainingFilePath);
            arffPanel2 = new ArffPanel(testFilePath.getPath());
            testArffScrollPane.setViewportView(arffPanel2);
            useTestSetRadio.setEnabled(true);
            testSet = arffLoad.getDataSet();
            printStatus("Test file: " + testFilePath.getPath() + " Loaded.");
        }
        catch (Exception e) {
            printStatus("Error loading test set: " + e.getMessage());
        }
    } 
    
    /** builds a Weka-compliant array of arguments for Evaluation
     * @param trainingFilename  name of the file containing the training set.
     * @param testFilename  name of the file containing the test set.
     * @param classAttribute  index of the class attribute (1,2,3 etc.)
     * @return an array of arguments
     */
    
    private String[] argumentBuilder(String trainingFilename, String testFilename, int classAttribute) {
        String[] args = new String[6];
        args[0] = "-t";
        args[1] = trainingFilename;
        args[2] = "-T";
        args[3] = testFilename;
        args[4] = "-c";
        args[5] = Integer.toString(classAttribute);
        return args;
    }
    
    
    private void runKnn() {
        try {
            testNumber++;
            if (trainingSet.classIndex() < 0) {
                    trainingSet.setClassIndex(trainingSet.numAttributes() - 1);
            }
            startButton.setEnabled(false);
            //Evaluation knnEval = new Evaluation(trainingSet);
            String output;
            int classInd;
            
            if (classifierSelect == 0) {
                KNN knn = new KNN(Integer.parseInt(kValueField.getText()));

                // this code crashes the program
               /* output = Evaluation.evaluateModel(knn, argumentBuilder(trainingFilePath.getPath(),
                        testFilePath.getPath(), trainingSet.classIndex()));
                */
                if (testSettingsSelect == 0) {
                    output = evaluate(knn, trainingSet, testSet);
                }
                else output = evaluate(knn, trainingSet, trainingSet);
                
                
                resultsTextArea.append("Test [" + testNumber + "] results: simple knn\n" +
                        "-------------\n" + output);