7 #include <dmlc/registry.h> 14 DMLC_REGISTRY_FILE_TAG(load_data_counts);
16 static void load_data_counts(ASTNode* node,
17 const std::vector<std::vector<size_t>>& counts) {
18 if (node->tree_id >= 0 && node->node_id >= 0) {
19 node->data_count = counts[node->tree_id][node->node_id];
21 for (ASTNode* child : node->children) {
22 load_data_counts(child, counts);
27 ASTBuilder::LoadDataCounts(
const std::vector<std::vector<size_t>>& counts) {
28 load_data_counts(this->main_node, counts);