Function GenerateFeaturesPermutation

catboost/libs/fstr/sage_values.cpp:205–228

};

TVector<std::pair<ui32, EFeatureType>> GenerateFeaturesPermutation(TFeaturesLayoutPtr featuresLayout,
                                                                   TRestorableFastRng64* randPtr) {
    TVector<std::pair<ui32, EFeatureType>> featuresPermutation;

    for (ui32 i = 0; i < featuresLayout->GetFloatFeatureCount(); ++i) {
        featuresPermutation.emplace_back(i, EFeatureType::Float);
    }

    for (ui32 i = 0; i < featuresLayout->GetCatFeatureCount(); ++i) {
        featuresPermutation.emplace_back(i, EFeatureType::Categorical);
    }

    for (ui32 i = 0; i < featuresLayout->GetTextFeatureCount(); ++i) {
        featuresPermutation.emplace_back(i, EFeatureType::Text);
    }

    for (ui32 i = 0; i < featuresLayout->GetEmbeddingFeatureCount(); ++i) {
        featuresPermutation.emplace_back(i, EFeatureType::Embedding);
    }

    Shuffle(featuresPermutation.begin(), featuresPermutation.end(), *randPtr);

    return featuresPermutation;
}

TDataProvider GetRandomDatasetBatch(
    const TDataProvider& dataset,