Function CopyTree

catboost/libs/train_interface/catboost_api.cpp:132–178

}

CATBOOST_API bool CopyTree(
    ResultHandle handle,
    int treeIndex,
    int* features,
    float* conditions,
    float* leaves,
    float* weights) {
    if (handle) {
        try {
            const auto modelTrees = RESULT_PTR(handle)->ModelTrees.GetMutable();

            size_t treeLeafCount = (1uLL << modelTrees->GetModelTreeData()->GetTreeSizes()[treeIndex]) * modelTrees->GetDimensionsCount();
            auto srcLeafValues = modelTrees->GetFirstLeafPtrForTree(treeIndex);
            const auto& srcWeights = modelTrees->GetModelTreeData()->GetLeafWeights();

            for (size_t idx = 0; idx < treeLeafCount; ++idx) {
                leaves[idx] = (float) srcLeafValues[idx];
            }

            auto applyData = modelTrees->GetApplyData();
            const size_t weightOffset = applyData->TreeFirstLeafOffsets[treeIndex] / modelTrees->GetDimensionsCount();
            for (size_t idx = 0; idx < (1uLL << modelTrees->GetModelTreeData()->GetTreeSizes()[treeIndex]); ++idx) {
                weights[idx] = (float) srcWeights[idx + weightOffset];
            }

            int treeSplitEnd;
            if (treeIndex + 1 < modelTrees->GetModelTreeData()->GetTreeStartOffsets().ysize()) {
                treeSplitEnd = modelTrees->GetModelTreeData()->GetTreeStartOffsets()[treeIndex + 1];
            } else {
                treeSplitEnd = modelTrees->GetModelTreeData()->GetTreeSplits().ysize();
            }
            const auto& binFeatures = modelTrees->GetBinFeatures();

            const auto offset = modelTrees->GetModelTreeData()->GetTreeStartOffsets()[treeIndex];
            for (int idx = offset; idx < treeSplitEnd; ++idx) {
                auto split = binFeatures[modelTrees->GetModelTreeData()->GetTreeSplits()[idx]];
                CB_ENSURE(split.Type == ESplitType::FloatFeature);
                features[idx - offset] = split.FloatFeature.FloatFeature;
                conditions[idx - offset] = split.FloatFeature.Split;
            }
        } catch (...) {
            Singleton<TErrorMessageHolder>()->Message = CurrentExceptionMessage();
            return false;
        }
    }
    return true;
}

CATBOOST_API bool TrainCatBoost(const TDataSet* trainPtr,
                                const TDataSet* testPtr,