/Classy

Simple kNN and NaiveBayes classifier implementation

Primary LanguageJavaMIT LicenseMIT

Classy Build Quality Gate Status

Simple kNN and NaiveBayes classifier implementation

Usage example

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