Method checkDataSet

JSAT/test/jsat/io/JSATDataTest.java:317–370

(DataSet ogData, DataSet cpData)

    }

    private void checkDataSet(DataSet ogData, DataSet cpData)
    {
        assertEquals(ogData.getClass().getCanonicalName(), cpData.getClass().getCanonicalName());
        
        assertEquals(ogData.getNumNumericalVars(), cpData.getNumNumericalVars());
        assertEquals(ogData.getNumCategoricalVars(), cpData.getNumCategoricalVars());
        assertEquals(ogData.getSampleSize(), cpData.getSampleSize());
        
        CategoricalData[] og_cats = ogData.getCategories();
        CategoricalData[] cp_cats = ogData.getCategories();
        
        for(int i = 0; i < og_cats.length; i++)
        {
            assertEquals(og_cats[i].getCategoryName(), cp_cats[i].getCategoryName());
            assertEquals(og_cats[i].getNumOfCategories(), cp_cats[i].getNumOfCategories());
            for(int j = 0; j < og_cats[i].getNumOfCategories(); j++)
                assertEquals(og_cats[i].getOptionName(j), cp_cats[i].getOptionName(j));
        }
        
        //compare datapoint values
        for(int i = 0; i < ogData.getSampleSize(); i++)
        {
            DataPoint og = ogData.getDataPoint(i);
            DataPoint cp = cpData.getDataPoint(i);
            
            assertArrayEquals(og.getCategoricalValues(), cp.getCategoricalValues());
            Vec og_vec = og.getNumericalValues();
            Vec cp_vec = cp.getNumericalValues();
            assertTrue(og_vec.isSparse() == cp_vec.isSparse());
            assertTrue(og_vec.equals(cp_vec));
            
            assertEquals(og.getWeight(), cp.getWeight(), 0.0);
            
        }
        
        if(ogData instanceof ClassificationDataSet)
        {
            ClassificationDataSet ogC = (ClassificationDataSet) ogData;
            ClassificationDataSet cpC = (ClassificationDataSet) cpData;
            
            for(int i = 0; i < ogData.getSampleSize(); i++)
                assertEquals(ogC.getDataPointCategory(i), cpC.getDataPointCategory(i));
        }
        
        if(ogData instanceof RegressionDataSet)
        {
            RegressionDataSet ogR = (RegressionDataSet) ogData;
            RegressionDataSet cpR = (RegressionDataSet) cpData;
            
            for(int i = 0; i < ogData.getSampleSize(); i++)
                assertEquals(ogR.getTargetValue(i), cpR.getTargetValue(i), 0.0);
        }
        
    }

    
}