Skip to content
Closed
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
43 changes: 43 additions & 0 deletions KNN.java
Original file line number Diff line number Diff line change
@@ -0,0 +1,43 @@
import java.io.BufferedReader;
import java.io.FileNotFoundException;
import java.io.FileReader;

import weka.classifiers.Classifier;
import weka.classifiers.lazy.IBk;
import weka.core.Instance;
import weka.core.Instances;

public class KNN {
public static BufferedReader readDataFile(String filename) {
BufferedReader inputReader = null;

try {
inputReader = new BufferedReader(new FileReader(filename));
} catch (FileNotFoundException ex) {
System.err.println("File not found: " + filename);
}

return inputReader;
}

public static void main(String[] args) throws Exception {
BufferedReader datafile = readDataFile("data.txt");

Instances data = new Instances(datafile);
data.setClassIndex(data.numAttributes() - 1);

//do not use first and second
Instance first = data.instance(0);
Instance second = data.instance(1);
data.delete(0);
data.delete(1);

Classifier ibk = new IBk();
ibk.buildClassifier(data);

double class1 = ibk.classifyInstance(first);
double class2 = ibk.classifyInstance(second);

System.out.println("first: " + class1 + "\nsecond: " + class2);
}
}