Method GetLearnFeatureIds

catboost/cuda/gpu_data/dataset_helpers.cpp:73–134

}

TVector<ui32> NCatboostCuda::GetLearnFeatureIds(NCatboostCuda::TBinarizedFeaturesManager& featuresManager) {
    TSet<ui32> featureIdsSet;
    auto ctrTypes = featuresManager.GetKnownSimpleCtrTypes();

    for (auto floatFeature : featuresManager.GetFloatFeatureIds()) {
        if (featuresManager.GetBinCount(floatFeature) > 1) {
            featureIdsSet.insert(floatFeature);
        }
    }
    for (auto catFeature : featuresManager.GetCatFeatureIds()) {
        if (featuresManager.UseForOneHotEncoding(catFeature)) {
            if (featuresManager.GetBinCount(catFeature) > 1) {
                featureIdsSet.insert(catFeature);
            }
        }

        if (featuresManager.UseForCtr(catFeature)) {
            for (auto& ctr : ctrTypes) {
                const auto simpleCtrsForType = featuresManager.CreateSimpleCtrsForType(catFeature,
                                                                                       ctr);
                for (auto ctrFeatureId : simpleCtrsForType) {
                    featureIdsSet.insert(ctrFeatureId);
                }
            }
        }
    }
    TSet<ui32> combinationCtrIds;

    for (auto& ctr : ctrTypes) {
        auto combinationCtrs = featuresManager.CreateCombinationCtrForType(ctr);
        for (auto ctrFeatureId : combinationCtrs) {
            TFeatureTensor tensor = featuresManager.GetCtr(ctrFeatureId).FeatureTensor;
            bool hasUnknownFeatures = false;
            CB_ENSURE(tensor.GetSplits().size() == 0);

            for (auto featureId : tensor.GetCatFeatures()) {
                if (!featureIdsSet.contains(featureId)) {
                    hasUnknownFeatures = true;
                    break;
                }
            }
            for (auto binarySplit : tensor.GetSplits()) {
                if (!featureIdsSet.contains(binarySplit.FeatureId)) {
                    hasUnknownFeatures = true;
                    break;
                }
            }
            if (!hasUnknownFeatures) {
                combinationCtrIds.insert(ctrFeatureId);
            }
        }
    }
    featureIdsSet.insert(combinationCtrIds.begin(), combinationCtrIds.end());

    auto estimatedFeatures = featuresManager.GetEstimatedFeatureIds();
    featureIdsSet.insert(estimatedFeatures.begin(), estimatedFeatures.end());

    auto featureBundleIds = featuresManager.GetExclusiveFeatureBundleIds();