Function GetTreeNodeToLeaf

catboost/private/libs/algo/tree_print.cpp:264–279

}

TVector<ui32> GetTreeNodeToLeaf(const TFullModel& model, size_t treeIdx) {
    CB_ENSURE(treeIdx < model.GetTreeCount(),
        "Requested tree leaves for nodes for tree " << treeIdx << ", but model has " << model.GetTreeCount());
    Y_ASSERT(!model.IsOblivious());
    const size_t offset = model.ModelTrees->GetModelTreeData()->GetTreeStartOffsets()[treeIdx];
    const auto start = model.ModelTrees->GetModelTreeData()->GetNonSymmetricNodeIdToLeafId().begin() + offset;
    const auto end = start + model.ModelTrees->GetModelTreeData()->GetTreeSizes()[treeIdx];
    auto applyData = model.ModelTrees->GetApplyData();
    const size_t firstLeafOffset = applyData->TreeFirstLeafOffsets[treeIdx];
    const size_t dimensionsCount = model.GetDimensionsCount();
    TVector<ui32> nodeToLeaf(start, end);
    for (auto& value : nodeToLeaf) {
        value = (value - firstLeafOffset) / dimensionsCount;
    }
    return nodeToLeaf;
}