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

Method PrepareForInitialization

catboost/libs/data/objects.cpp:182–213  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

180}
181
182void NCB::TCommonObjectsData::PrepareForInitialization(
183 const TDataMetaInfo& metaInfo,
184 ui32 objectCount,
185 ui32 prevTailCount
186) {
187 FeaturesLayout = metaInfo.FeaturesLayout;
188 if (prevTailCount == 0) {
189 SetStoreStringColumns(metaInfo.StoreStringColumns);
190 } else {
191 SubgroupIds.SetStoreStringColumnsVal(metaInfo.StoreStringColumns);
192 GroupIds.SetStoreStringColumnsVal(metaInfo.StoreStringColumns);
193 StoreStringColumns = metaInfo.StoreStringColumns;
194 }
195
196 if (StoreStringColumns) {
197 NCB::PrepareForInitialization(metaInfo.HasGroupId, objectCount, prevTailCount, &GroupIds.GetMaybeStringData());
198 NCB::PrepareForInitialization(metaInfo.HasSubgroupIds, objectCount, prevTailCount, &SubgroupIds.GetMaybeStringData());
199 NCB::PrepareForInitialization(metaInfo.HasSampleId, objectCount, prevTailCount, &SampleId);
200 } else {
201 NCB::PrepareForInitialization(metaInfo.HasGroupId, objectCount, prevTailCount, &GroupIds.GetMaybeNumData());
202 NCB::PrepareForInitialization(metaInfo.HasSubgroupIds, objectCount, prevTailCount, &SubgroupIds.GetMaybeNumData());
203 }
204 NCB::PrepareForInitialization(metaInfo.HasTimestamp, objectCount, prevTailCount, &Timestamp);
205
206 const size_t catFeatureCount = (size_t)metaInfo.FeaturesLayout->GetCatFeatureCount();
207 if (catFeatureCount) {
208 if (!CatFeaturesHashToString) {
209 CatFeaturesHashToString = MakeAtomicShared<TVector<THashMap<ui32, TString>>>();
210 }
211 CatFeaturesHashToString->resize(catFeatureCount);
212 }
213}
214
215void NCB::TCommonObjectsData::SetBuildersArrayRef(
216 const TDataMetaInfo& metaInfo,

Callers

nothing calls this directly

Calls 8

PrepareForInitializationFunction · 0.85
GetCatFeatureCountMethod · 0.80
resizeMethod · 0.45
clearMethod · 0.45
GetFloatFeatureCountMethod · 0.45
GetTextFeatureCountMethod · 0.45

Tested by

no test coverage detected