| 193 | } |
| 194 | |
| 195 | int32 BoostedTreesEnsembleResource::right_id(const int32 tree_id, |
| 196 | const int32 node_id) const { |
| 197 | const auto node = tree_ensemble_->trees(tree_id).nodes(node_id); |
| 198 | DCHECK_EQ(node.node_case(), boosted_trees::Node::kBucketizedSplit); |
| 199 | return node.bucketized_split().right_id(); |
| 200 | } |
| 201 | |
| 202 | std::vector<float> BoostedTreesEnsembleResource::GetTreeWeights() const { |
| 203 | return {tree_ensemble_->tree_weights().begin(), |