@Test public void testConstructFromProbability() { float[] probability = {0.0f, 0.125f, 0.375f, 0.0f, 0.5f, 0.0f}; CategoricalPrediction prediction = new CategoricalPrediction(probability); assertEquals(FeatureType.CATEGORICAL, prediction.getFeatureType()); assertEquals(4, prediction.getMostProbableCategoryID()); assertNull(prediction.getCategoryCounts()); assertArrayEquals(probability, prediction.getCategoryProbabilities()); }
@Test public void testConstruct() { int[] counts = {0, 1, 3, 0, 4, 0}; CategoricalPrediction prediction = new CategoricalPrediction(counts); assertEquals(FeatureType.CATEGORICAL, prediction.getFeatureType()); assertEquals(4, prediction.getMostProbableCategoryID()); assertArrayEquals(counts, prediction.getCategoryCounts()); assertArrayEquals( new float[] {0.0f, 0.125f, 0.375f, 0.0f, 0.5f, 0.0f}, prediction.getCategoryProbabilities()); }