Класс вероятности DecisionTree в млпаке

Question

У меня есть дерево решений в mlpackи метки 0 или 1. Я тренирую дерево, используя Train(TrainingSet, TrainingLabels, 2, 1, 1e-6)где 2 - количество классов. (тренировочные данные также векторы 0,1)

Я классифицирую n контрольных точек, используя Classify<>(TestData, Predictions, Probabilities),

Матрица вероятностей содержит числа 2 на n. Как я могу определить, какой ряд соответствует какому классу?