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

Method FillGroupOrder

catboost/cuda/data/permutation.cpp:20–28  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

18}
19
20void NCatboostCuda::TDataPermutation::FillGroupOrder(TVector<ui32>& groupOrder) const {
21 CB_ENSURE_INTERNAL(DataProvider->MetaInfo.HasGroupId, "FillGroupOrder supports only datasets with group ids");
22 if (Index != IdentityPermutationId()) {
23 Shuffle(GetSeed(), BlockSize, DataProvider->ObjectsGrouping->GetGroupCount(), &groupOrder);
24 } else {
25 groupOrder.yresize(DataProvider->ObjectsGrouping->GetGroupCount());
26 std::iota(groupOrder.begin(), groupOrder.end(), 0);
27 }
28}
29
30void NCatboostCuda::TDataPermutation::FillInversePermutation(TVector<ui32>& permutation) const {
31 TVector<ui32> order;

Callers 1

TQueriesGroupingMethod · 0.80

Calls 5

GetGroupCountMethod · 0.80
yresizeMethod · 0.80
ShuffleFunction · 0.70
beginMethod · 0.45
endMethod · 0.45

Tested by

no test coverage detected