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

Function GetMaybeSubsetDataProvider

catboost/libs/data/ut/objects_ut.cpp:30–53  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

28
29template <class TTObjectsDataProvider>
30static TTObjectsDataProvider GetMaybeSubsetDataProvider(
31 TTObjectsDataProvider&& objectsDataProvider,
32 TMaybe<TArraySubsetIndexing<ui32>> subsetForGetSubset,
33 TMaybe<EObjectsOrder> objectsOrderForGetSubset,
34 NPar::ILocalExecutor* localExecutor
35) {
36 if (subsetForGetSubset.Defined()) {
37 TObjectsGroupingSubset objectsGroupingSubset = GetSubset(
38 objectsDataProvider.GetObjectsGrouping(),
39 TArraySubsetIndexing<ui32>(*subsetForGetSubset),
40 *objectsOrderForGetSubset
41 );
42
43 auto subsetDataProvider = objectsDataProvider.GetSubset(
44 objectsGroupingSubset,
45 NSystemInfo::TotalMemorySize(),
46 localExecutor
47 );
48 objectsDataProvider = std::move(
49 dynamic_cast<TTObjectsDataProvider&>(*subsetDataProvider.Get())
50 );
51 }
52 return std::move(objectsDataProvider);
53}
54
55template <class T>
56bool Equal(TMaybeData<TConstArrayRef<T>> lhs, TMaybeData<TVector<T>> rhs) {

Callers 1

TestSubsetFeaturesFunction · 0.85

Calls 6

GetSubsetFunction · 0.85
moveFunction · 0.50
DefinedMethod · 0.45
GetObjectsGroupingMethod · 0.45
GetSubsetMethod · 0.45
GetMethod · 0.45

Tested by 1

TestSubsetFeaturesFunction · 0.68