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

Function SetDataDependentDefaultsForGpu

catboost/cuda/train_lib/train.cpp:201–210  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

199 }
200
201 static void SetDataDependentDefaultsForGpu(const NCB::TTrainingDataProvider& dataProvider,
202 const NCB::TTrainingDataProvider* testProvider,
203 NCatboostOptions::TCatBoostOptions& catBoostOptions,
204 TBinarizedFeaturesManager& featuresManager,
205 NPar::ILocalExecutor* localExecutor) {
206 UpdateGpuSpecificDefaults(catBoostOptions, featuresManager);
207 EstimatePriors(dataProvider, featuresManager, catBoostOptions.CatFeatureParams, localExecutor);
208 UpdateDataPartitionType(featuresManager, catBoostOptions);
209 UpdatePinnedMemorySizeOption(dataProvider, testProvider, featuresManager, catBoostOptions);
210 }
211
212 static void ConfigureCudaProfiler(bool isProfile, NCudaLib::TCudaProfiler* profiler) {
213 if (isProfile) {

Callers 2

TrainModelMethod · 0.85
ModelBasedEvalMethod · 0.85

Calls 4

EstimatePriorsFunction · 0.85
UpdateDataPartitionTypeFunction · 0.85

Tested by

no test coverage detected