Function GetSubsetForFstrCalc

catboost/libs/fstr/loss_change_fstr.cpp:88–120

}

const TDataProviderPtr GetSubsetForFstrCalc(
    const TDataProviderPtr dataset,
    NPar::ILocalExecutor* localExecutor)
{
    ui32 totalDocumentCount = dataset->ObjectsData->GetObjectCount();
    ui32 maxDocumentCount = SafeIntegerCast<ui32>(
        GetMaxObjectCountForFstrCalc(
            totalDocumentCount,
            SafeIntegerCast<i64>(dataset->ObjectsData->GetFeaturesLayout()->GetExternalFeatureCount())
        )
    );

    if (totalDocumentCount > maxDocumentCount) {
        ui32 foldCount = totalDocumentCount / maxDocumentCount;

        TVector<NCB::TArraySubsetIndexing<ui32>> testSubsets;

        testSubsets = NCB::Split(*dataset->ObjectsGrouping, foldCount, /*oldCvStyleSplit*/ true);

        auto subset = dataset->GetSubset(
            GetSubset(
                dataset->ObjectsGrouping,
                std::move(testSubsets[0]),
                NCB::EObjectsOrder::Ordered
            ),
            NSystemInfo::TotalMemorySize(),
            localExecutor
        );
        return subset;
    } else {
        return dataset;
    }
}

void CreateMetricAndLossDescriptionForLossChange(
    const TFullModel& model,