MCPcopy Create free account
hub / github.com/catboost/catboost / GetCatFeatureIds

Method GetCatFeatureIds

catboost/cuda/data/binarizations_manager.cpp:319–332  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

317 }
318
319 TVector<ui32> TBinarizedFeaturesManager::GetCatFeatureIds() const {
320 const auto& featuresLayout = *QuantizedFeaturesInfo->GetFeaturesLayout();
321 const auto& metaInfo = featuresLayout.GetExternalFeaturesMetaInfo();
322
323 TVector<ui32> featureIds;
324
325 for (const auto& feature : DataProviderCatFeatureIdToFeatureManagerId) {
326 if (metaInfo[feature.first].IsAvailable && !IgnoredFeatures.contains(feature.first)) {
327 featureIds.push_back(feature.second);
328 }
329 }
330 Sort(featureIds.begin(), featureIds.end());
331 return featureIds;
332 }
333
334 TVector<ui32> TBinarizedFeaturesManager::GetFloatFeatureIds() const {
335 const auto& featuresLayout = *QuantizedFeaturesInfo->GetFeaturesLayout();

Calls 7

SortFunction · 0.50
GetFeaturesLayoutMethod · 0.45
containsMethod · 0.45
push_backMethod · 0.45
beginMethod · 0.45
endMethod · 0.45

Tested by 2

TestTreeBuilderFunction · 0.64