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

Method GetFloatFeatureIds

catboost/cuda/data/binarizations_manager.cpp:334–346  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

332 }
333
334 TVector<ui32> TBinarizedFeaturesManager::GetFloatFeatureIds() const {
335 const auto& featuresLayout = *QuantizedFeaturesInfo->GetFeaturesLayout();
336 const auto& metaInfo = featuresLayout.GetExternalFeaturesMetaInfo();
337
338 TVector<ui32> featureIds;
339
340 for (const auto& feature : DataProviderFloatFeatureIdToFeatureManagerId) {
341 if (metaInfo[feature.first].IsAvailable && !IgnoredFeatures.contains(feature.first)) {
342 featureIds.insert(featureIds.end(), feature.second.begin(), feature.second.end());
343 }
344 }
345 return featureIds;
346 }
347
348 TVector<ui32> TBinarizedFeaturesManager::GetDataProviderFeatureIds() const {
349 TVector<ui32> features;

Callers 4

GetLearnFeatureIdsMethod · 0.80
TestTreeBuilderFunction · 0.80
TrainModelMethod · 0.80

Calls 6

GetFeaturesLayoutMethod · 0.45
containsMethod · 0.45
insertMethod · 0.45
endMethod · 0.45
beginMethod · 0.45

Tested by 2

TestTreeBuilderFunction · 0.64