185186187188189190191192193194195
trees.add(value.getTree()); int[] predictions = value.getPredictions(); for (int id = 0; id < predictions.length; id++) { callback.prediction(key.treeId(), firstIds[key.partition()] + id, predictions[id]); } } return new DecisionForest(trees);
135136137138139140141142143144145
} TreeID k = output.getKeys()[current]; // the tree should receive the partition's index assertEquals(partition, k.partition()); // make sure all the trees of the other partitions are handled in the // correct order assertEquals(index, k.treeId());
142143144145146147148149150151152