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

Function UpdateDataPartitionType

catboost/cuda/train_lib/train.cpp:73–84  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

71 }
72
73 inline void UpdateDataPartitionType(const TBinarizedFeaturesManager& featuresManager,
74 NCatboostOptions::TCatBoostOptions& catBoostOptions) {
75 if (catBoostOptions.CatFeatureParams->MaxTensorComplexity > 1 && featuresManager.GetCatFeatureIds().size()) {
76 return;
77 } else {
78 if (catBoostOptions.BoostingOptions->BoostingType == EBoostingType::Plain) {
79 if (catBoostOptions.BoostingOptions->DataPartitionType.NotSet()) {
80 catBoostOptions.BoostingOptions->DataPartitionType = EDataPartitionType::DocParallel;
81 }
82 }
83 }
84 }
85
86 inline bool HasPermutationFeatures(const TBinarizedFeaturesManager& featuresManager) {
87 for (auto catFeature : featuresManager.GetCatFeatureIds()) {

Callers 1

Calls 3

GetCatFeatureIdsMethod · 0.80
NotSetMethod · 0.80
sizeMethod · 0.45

Tested by

no test coverage detected