Method ProcessSplitsSet

catboost/libs/model/model_build_helper.cpp:100–126

}

void TCommonModelBuilderHelper::ProcessSplitsSet(const TSet<TModelSplit>& modelSplitSet, TModelTrees* trees) {
    trees->SetApproxDimension(ApproxDimension);
    for (auto& feature : FloatFeatures) {
        feature.Borders.clear();
    }
    trees->SetFloatFeatures(std::move(FloatFeatures));

    MakeFeaturesUnused(MakeArrayRef(CatFeatures.begin(), CatFeatures.end()));
    MakeFeaturesUnused(MakeArrayRef(TextFeatures.begin(), TextFeatures.end()));
    MakeFeaturesUnused(MakeArrayRef(EmbeddingFeatures.begin(), EmbeddingFeatures.end()));

    trees->SetCatFeatures(std::move(CatFeatures));
    trees->SetTextFeatures(std::move(TextFeatures));
    trees->SetEmbeddingFeatures(std::move(EmbeddingFeatures));

    trees->ProcessSplitsSet(modelSplitSet,
                            FloatFeaturesInternalIndexesMap,
                            CatFeaturesInternalIndexesMap,
                            TextFeaturesInternalIndexesMap,
                            EmbeddingFeaturesInternalIndexesMap);
    for (const auto& split : modelSplitSet) {
        const int binFeatureIdx = BinFeatureIndexes.ysize();
        Y_ASSERT(!BinFeatureIndexes.contains(split));
        BinFeatureIndexes[split] = binFeatureIdx;
    }
    Y_ASSERT(modelSplitSet.size() == BinFeatureIndexes.size());
}


TObliviousTreeBuilder::TObliviousTreeBuilder(