Package quickml.supervised.classifier.decisionTree

Source Code of quickml.supervised.classifier.decisionTree.ClassificationCounterTest

package quickml.supervised.classifier.decisionTree;

import org.testng.Assert;
import org.testng.annotations.Test;
import quickml.supervised.classifier.decisionTree.tree.ClassificationCounter;

/**
* Created by ian on 2/27/14.
*/
public class ClassificationCounterTest {

    @Test
    public void testAdd() {
        ClassificationCounter a = new ClassificationCounter();
        a.addClassification("dog", 1.0);
        a.addClassification("cat", 0.5);
        ClassificationCounter b = new ClassificationCounter();
        b.addClassification("dog", 0.5);
        b.addClassification("cat", 1.0);
        ClassificationCounter c = a.add(b);
        Assert.assertEquals(c.getCount("dog"), 1.5);
        Assert.assertEquals(c.getCount("cat"), 1.5);
    }

    @Test
    public void testSubtract() {
        ClassificationCounter a = new ClassificationCounter();
        a.addClassification("dog", 1.0);
        a.addClassification("cat", 2.5);
        ClassificationCounter b = new ClassificationCounter();
        b.addClassification("dog", 0.5);
        b.addClassification("cat", 1.0);
        ClassificationCounter c = a.subtract(b);
        Assert.assertEquals(c.getCount("dog"), 0.5);
        Assert.assertEquals(c.getCount("cat"), 1.5);
    }

    @Test
    public void testMerge() {
        ClassificationCounter a = new ClassificationCounter();
        a.addClassification("dog", 1.0);
        a.addClassification("cat", 0.5);
        ClassificationCounter b = new ClassificationCounter();
        b.addClassification("dog", 0.5);
        b.addClassification("cat", 1.0);
        ClassificationCounter merged = ClassificationCounter.merge(a, b);
        Assert.assertEquals(merged.getTotal(), 3.0);
        Assert.assertEquals(merged.getCount("dog"), 1.5);
        Assert.assertEquals(merged.getCount("cat"), 1.5);
    }
}
TOP

Related Classes of quickml.supervised.classifier.decisionTree.ClassificationCounterTest

TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.