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

Function SetPairs

catboost/libs/data/loader.cpp:179–193  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

177 }
178
179 void SetPairs(const TPathWithScheme& pairsPath, TDatasetSubset loadSubset, IDatasetVisitor* visitor) {
180 DumpMemUsage("After data read");
181 if (pairsPath.Inited()) {
182 auto pairsDataLoader = GetProcessor<IPairsDataLoader>(
183 pairsPath,
184 TPairsDataLoaderArgs{pairsPath, loadSubset}
185 );
186 if (pairsDataLoader->NeedGroupIdToIdxMap()) {
187 auto maybeGroupIds = visitor->GetGroupIds();
188 CB_ENSURE(maybeGroupIds, "Cannot load pairs data with group ids for a dataset without groups");
189 pairsDataLoader->SetGroupIdToIdxMap(*maybeGroupIds);
190 }
191 pairsDataLoader->Do(visitor);
192 }
193 }
194
195 void SetGraph(const TPathWithScheme& graphPath, TDatasetSubset loadSubset, IDatasetVisitor* visitor) {
196 DumpMemUsage("Before data read graphPath");

Callers 3

DoMethod · 0.85
SetPairsMethod · 0.85
FinalizeBuilderMethod · 0.85

Calls 6

DumpMemUsageFunction · 0.85
InitedMethod · 0.45
NeedGroupIdToIdxMapMethod · 0.45
GetGroupIdsMethod · 0.45
SetGroupIdToIdxMapMethod · 0.45
DoMethod · 0.45

Tested by

no test coverage detected