У меня есть дерево решений в mlpack, а метки - 0 или 1. Я обучаю дерево, используя Train(TrainingSet, TrainingLabels, 2, 1, 1e-6), где 2 - количество классов.(обучающие данные также имеют векторы 0,1)
Я классифицирую n контрольных точек, используя Classify<>(TestData, Predictions, Probabilities).
Матрица вероятностей содержит числа 2 на n.Как я могу определить, какая строка соответствует какому классу?