Method GetSubset

catboost/libs/data/target.cpp:644–714

TRawTargetDataProvider TRawTargetDataProvider::GetSubset(
    const TObjectsGroupingSubset& objectsGroupingSubset,
    NPar::ILocalExecutor* localExecutor
) const {
    const TArraySubsetIndexing<ui32>& objectsSubsetIndexing = objectsGroupingSubset.GetObjectsIndexing();

    TRawTargetData subsetData;
    subsetData.TargetType = Data.TargetType;

    TVector<std::function<void()>> tasks;

    if (TMaybeData<TConstArrayRef<TRawTarget>> maybeTarget = GetTarget()) {
        TConstArrayRef<TRawTarget> target = *maybeTarget;
        subsetData.Target.resize(target.size());
        for (auto targetIdx : xrange(target.size())) {
            tasks.emplace_back(
                [&, target, targetIdx]() {
                    GetRawTargetSubset(
                        target[targetIdx],
                        objectsSubsetIndexing,
                        localExecutor,
                        &subsetData.Target[targetIdx]
                    );
                }
            );
        }
    }

    tasks.emplace_back(
        [&, this]() {
            GetMultidimBaselineSubset(
                Data.Baseline,
                objectsSubsetIndexing,
                localExecutor,
                &subsetData.Baseline
            );
        }
    );

    tasks.emplace_back(
        [&, this]() {
            subsetData.Weights = Data.Weights.GetSubset(objectsSubsetIndexing, localExecutor);
        }
    );

    tasks.emplace_back(
        [&, this]() {
            subsetData.GroupWeights = Data.GroupWeights.GetSubset(objectsSubsetIndexing, localExecutor);
        }
    );

    if (Data.Pairs) {
        tasks.emplace_back(
            [&, this]() {
                TRawPairsData subsetPairs;
                GetPairsSubset(*Data.Pairs, *ObjectsGrouping, objectsGroupingSubset, &subsetPairs);
                subsetData.Pairs = std::move(subsetPairs);
            }