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

Method CreateDataSet

catboost/cuda/methods/dynamic_boosting.h:141–157  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

139 }
140
141 TFeatureParallelDataSetsHolder CreateDataSet() const {
142 CB_ENSURE(DataProvider);
143 ui32 permutationBlockSize = GetPermutationBlockSize(DataProvider->GetObjectCount());
144
145 TFeatureParallelDataSetHoldersBuilder dataSetsHolderBuilder(FeaturesManager,
146 *DataProvider,
147 *Estimators,
148 TestDataProvider,
149 permutationBlockSize,
150 CatFeaturesStorage
151
152 );
153
154 const auto permutationCount = DataProvider->ObjectsData->GetOrder() == NCB::EObjectsOrder::Ordered ? 1
155 : Config.PermutationCount;
156 return dataSetsHolderBuilder.BuildDataSet(permutationCount, LocalExecutor);
157 }
158
159 TPermutationTarget CreateTargets(const TFeatureParallelDataSetsHolder& dataSets) const {
160 TVector<THolder<TObjective>> targets;

Callers

nothing calls this directly

Calls 3

GetOrderMethod · 0.80
GetObjectCountMethod · 0.45
BuildDataSetMethod · 0.45

Tested by

no test coverage detected