Method AddBinTree

catboost/libs/model/model.cpp:238–249

}

void TModelTrees::AddBinTree(const TVector<int>& binSplits) {
    auto& data = *CastToSolidTree(*this);

    Y_ASSERT(data.TreeSizes.size() == data.TreeStartOffsets.size() && data.TreeSplits.empty() == data.TreeSizes.empty());
    data.TreeSplits.insert(data.TreeSplits.end(), binSplits.begin(), binSplits.end());
    if (data.TreeStartOffsets.empty()) {
        data.TreeStartOffsets.push_back(0);
    } else {
        data.TreeStartOffsets.push_back(data.TreeStartOffsets.back() + data.TreeSizes.back());
    }
    data.TreeSizes.push_back(binSplits.ysize());
}

void TModelTrees::ClearLeafWeights() {
    CastToSolidTree(*this)->LeafWeights.clear();