knnTest

knnTest.java

package br.com.quintoandar.classifier;

import lombok.val;

import net.sf.javaml.core.Dataset;
import net.sf.javaml.core.DefaultDataset;
import net.sf.javaml.core.Instance;
import net.sf.javaml.distance.CosineDistance;
import net.sf.javaml.tools.InstanceTools;
import org.junit.Test;

public class ClassifierTest {

    @Test
    public void test() {
        Dataset data = new DefaultDataset();
        for (int i = 0; i < 10000; i++) {
            Instance tmpInstance = InstanceTools.randomInstance(64);
            data.add(tmpInstance);
        }

        Instance testInstance = InstanceTools.randomInstance(64);

        val result = data.kNearest(5, testInstance, new CosineDistance());

        System.out.println(result);
    }
}