Simple kNN and NaiveBayes classifier implementation
public class Test {
public static void main(String[] args) {
PrintStream out = System.out;
BaseLoader csvLoader = new CsvLoader();
BaseClassifier knn = new KNearestNeighbor(4);
ClassificationRunner knnRunner = new ClassificationRunner(knn);
csvLoader.loadInput(FileInputReaderFactory.make("data/glass/glass.csv"));
InstanceSet glassDataset = InstanceSetFactory.make(csvLoader, "glass dataset");
try {
glassDataset.dropFields(Arrays.asList("Al", "K"));
} catch (UndefinedFieldException e) {
// handle error
}
out.println(String.format("KNN: %f", knnRunner.crossValidate(glassDataset, 10)));
}
}
Example project see on example
directory