mlpack中DecisionTree的类概率

Question

我在mlpack中有一个决策树，标签为0或1。 我使用Train(TrainingSet, TrainingLabels, 2, 1, 1e-6)训练树 其中2是班级数量。 （训练数据也是0,1的向量）

我使用Classify<>(TestData, Predictions, Probabilities)对n个测试点进行了分类。

概率矩阵包含2乘n的数字。我怎么知道哪一行对应哪个班级？