Method SetSeparateInitModel

catboost/private/libs/algo/learn_context.cpp:637–716

void TLearnProgress::SetSeparateInitModel(
    const TFullModel& initModel,
    const TDataProviders& initModelApplyCompatiblePools,
    bool isOrderedBoosting,
    bool storeExpApproxes,
    NPar::ILocalExecutor* localExecutor) {

    CATBOOST_DEBUG_LOG << "TLearnProgress::SetSeparateInitModel\n";

    SeparateInitModelTreesSize = SafeIntegerCast<ui32>(initModel.GetTreeCount());
    SeparateInitModelCheckSum = CalcCoreModelCheckSum(initModel);

    // Calc approxes

    auto calcApproxFunction = [&] (const TDataProvider& data) -> TVector<TVector<double>> {
        return ApplyModelMulti(
            initModel,
            *data.ObjectsData,
            EPredictionType::RawFormulaVal,
            0,
            SafeIntegerCast<int>(initModel.GetTreeCount()),
            localExecutor,
            data.RawTargetData.GetBaseline()
        );
    };

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

    tasks.push_back(
        [&] () {
            const ui32 learnObjectCount = initModelApplyCompatiblePools.Learn->GetObjectCount();
            if (!learnObjectCount) {
                return;
            }

            AvrgApprox = calcApproxFunction(*initModelApplyCompatiblePools.Learn);

            TVector<TConstArrayRef<double>> approxRef(AvrgApprox.begin(), AvrgApprox.end());

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

            auto setFoldApproxes = [&] (bool isPlainFold, TFold* fold) {
                for (auto& bodyTail : fold->BodyTailArr) {
                    InitApproxFromBaseline(
                        isPlainFold ? learnObjectCount : SafeIntegerCast<ui32>(bodyTail.TailFinish),
                        TConstArrayRef<TConstArrayRef<double>>(approxRef),
                        fold->GetLearnPermutationArray(),
                        storeExpApproxes,
                        &bodyTail.Approx);
                }
            };

            for (auto foldIdx : xrange(Folds.size())) {
                tasks.push_back(
                    [&, foldIdx] () {
                        setFoldApproxes(!isOrderedBoosting, &Folds[foldIdx]);
                    }
                );