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

Method CreateSimpleCtrsForType

catboost/cuda/data/binarizations_manager.cpp:217–233  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

215 }
216
217 TVector<ui32> TBinarizedFeaturesManager::CreateSimpleCtrsForType(ui32 featureId, ECtrType type) {
218 CB_ENSURE(UseForCtr(featureId));
219 TSet<ui32> resultIds;
220
221 if (HasPerFeatureCtr(featureId)) {
222 auto perFeatureCtrs = CreateGrouppedPerFeatureCtr(featureId);
223 if (perFeatureCtrs.contains(type)) {
224 CreateSimpleCtrs(featureId, perFeatureCtrs.at(type), &resultIds);
225 }
226 } else {
227 auto simpleCtrConfigs = CreateGrouppedSimpleCtrConfigs();
228 CB_ENSURE(simpleCtrConfigs.at(type), "Simple ctr type is not enabled " << type);
229 CreateSimpleCtrs(featureId, simpleCtrConfigs.at(type), &resultIds);
230 }
231
232 return TVector<ui32>(resultIds.begin(), resultIds.end());
233 }
234
235 TVector<ui32> TBinarizedFeaturesManager::GetAllSimpleCtrs() const {
236 TVector<ui32> ids;

Callers 1

GetLearnFeatureIdsMethod · 0.80

Calls 4

containsMethod · 0.45
atMethod · 0.45
beginMethod · 0.45
endMethod · 0.45

Tested by

no test coverage detected