Method testTrain_C

JSAT/test/jsat/classifiers/linear/ROMMATest.java:49–70

Test of supportsWeightedData method, of class ROMMA.

()

     * Test of supportsWeightedData method, of class ROMMA.
     */
    @Test
    public void testTrain_C()
    {
        System.out.println("supportsWeightedData");
        ROMMA nonAggro = new ROMMA();
        ROMMA aggro = new ROMMA();
        ClassificationDataSet train = FixedProblems.get2ClassLinear(200, RandomUtil.getRandom());
        
        nonAggro.setEpochs(1);
        nonAggro.trainC(train);
        
        aggro.setEpochs(1);
        aggro.trainC(train);
        
        ClassificationDataSet test = FixedProblems.get2ClassLinear(200, RandomUtil.getRandom());
        
        for(DataPointPair<Integer> dpp : test.getAsDPPList())
            assertEquals(dpp.getPair().longValue(), aggro.classify(dpp.getDataPoint()).mostLikely());
        
        for(DataPointPair<Integer> dpp : test.getAsDPPList())
            assertEquals(dpp.getPair().longValue(), nonAggro.classify(dpp.getDataPoint()).mostLikely());
    }
}