@Test
public void testPredictWithTwoClasses() {

    double first = 1;
    double second = 2;
    double[][] means = new double[][] {
        {5.855, 176.25, 11.25},
        {5.4175, 132.5, 7.5},
    };
    double[][] variances = new double[][] {
        {3.5033E-2, 1.2292E2, 9.1667E-1},
        {9.7225E-2, 5.5833E2, 1.6667},
    };
    double[] probabilities = new double[] {.5, .5};
    GaussianNaiveBayesModel mdl = new GaussianNaiveBayesModel(means, variances, probabilities, new double[] {first, second}, null);
    Vector observation = VectorUtils.of(6, 130, 8);
    Assert.assertEquals(second, mdl.predict(observation), 0.0001);
}