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

Function LoadPools

catboost/app/mode_select_features.cpp:72–113  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

70
71
72static TDataProviders LoadPools(
73 const TCatBoostOptions& catBoostOptions,
74 const TPoolLoadParams& poolLoadParams,
75 TOption<bool>* hasTimeFlag,
76 NPar::ILocalExecutor* executor
77) {
78 TVector<NJson::TJsonValue> classLabels = catBoostOptions.DataProcessingOptions->ClassLabels;
79 const auto objectsOrder = hasTimeFlag->Get() ? EObjectsOrder::Ordered : EObjectsOrder::Undefined;
80 CB_ENSURE(poolLoadParams.TestSetPaths.size() <= 1, "Features selection mode doesn't support several eval sets.");
81 const bool haveLearnFeaturesInMemory = HaveFeaturesInMemory(
82 catBoostOptions,
83 poolLoadParams.LearnSetPath);
84 TVector<TDatasetSubset> testDatasetSubsets;
85 for (const auto& testSetPath : poolLoadParams.TestSetPaths) {
86 testDatasetSubsets.push_back(
87 TDatasetSubset::MakeColumns(HaveFeaturesInMemory(catBoostOptions, testSetPath)));
88 }
89 auto pools = NCB::ReadTrainDatasets(
90 catBoostOptions.GetTaskType(),
91 poolLoadParams,
92 objectsOrder,
93 /*readTestData*/true,
94 TDatasetSubset::MakeColumns(haveLearnFeaturesInMemory),
95 testDatasetSubsets,
96 catBoostOptions.DataProcessingOptions->ForceUnitAutoPairWeights,
97 &classLabels,
98 executor,
99 /*profile*/nullptr
100 );
101 CB_ENSURE(pools.Learn != nullptr, "Train data must be provided");
102
103 const auto learnDataOrder = pools.Learn->ObjectsData->GetOrder();
104 if (learnDataOrder == EObjectsOrder::Ordered) {
105 *hasTimeFlag = true;
106 }
107 pools.Learn = ReorderByTimestampLearnDataIfNeeded(catBoostOptions, pools.Learn, executor);
108
109 TRestorableFastRng64 rand(catBoostOptions.RandomSeed);
110 pools.Learn = ShuffleLearnDataIfNeeded(catBoostOptions, pools.Learn, executor, &rand);
111
112 return pools;
113}
114
115
116void SaveSummaryToFile(const TFeaturesSelectionSummary& summary, const TString& resultPath) {

Callers 1

mode_select_featuresFunction · 0.70

Calls 9

HaveFeaturesInMemoryFunction · 0.85
ReadTrainDatasetsFunction · 0.85
ShuffleLearnDataIfNeededFunction · 0.85
GetTaskTypeMethod · 0.80
GetOrderMethod · 0.80
GetMethod · 0.45
sizeMethod · 0.45
push_backMethod · 0.45

Tested by

no test coverage detected