Function SetDataDependentDefaults

catboost/libs/train_lib/options_helper.cpp:403–435

}

void SetDataDependentDefaults(
    const NCB::TDataMetaInfo& trainDataMetaInfo,
    const TMaybe<NCB::TDataMetaInfo>& testDataMetaInfo,
    bool continueFromModel,
    bool continueFromProgress,
    NCatboostOptions::TOutputFilesOptions* outputFilesOptions,
    NCatboostOptions::TCatBoostOptions* catBoostOptions
) {
    const ui64 learnPoolSize = trainDataMetaInfo.ObjectCount;
    const ui64 testPoolSize = testDataMetaInfo.Defined() ? testDataMetaInfo->ObjectCount : 0;
    const bool isConstTestTarget = testDataMetaInfo.Defined() && IsConstTarget(*testDataMetaInfo);
    const bool hasTestPairs = testDataMetaInfo.Defined() && testDataMetaInfo->HasPairs;
    UpdateUseBestModel(testPoolSize, isConstTestTarget, hasTestPairs, outputFilesOptions);
    UpdateBoostingTypeOption(learnPoolSize, catBoostOptions);
    AdjustBoostFromAverageDefaultValue(trainDataMetaInfo, testDataMetaInfo, continueFromModel, catBoostOptions);
    UpdateLearningRate(learnPoolSize, outputFilesOptions->UseBestModel.Get(), catBoostOptions);
    UpdateOneHotMaxSize(
        trainDataMetaInfo.MaxCatFeaturesUniqValuesOnLearn,
        trainDataMetaInfo.TargetCount > 0,
        catBoostOptions
    );
    UpdateYetiRankEvalMetric(
        trainDataMetaInfo.TargetStats,
        testDataMetaInfo.Defined() ? testDataMetaInfo->TargetStats : Nothing(),
        catBoostOptions
    );
    UpdateLeavesEstimationIterations(trainDataMetaInfo, catBoostOptions);
    UpdateAndValidateMonotoneConstraints(*trainDataMetaInfo.FeaturesLayout.Get(), catBoostOptions);
    DropModelShrinkageIfBaselineUsed(trainDataMetaInfo, continueFromModel || continueFromProgress, catBoostOptions);
    UpdateDictionaryDefaults(learnPoolSize, catBoostOptions);
    UpdateSampleRateOption(learnPoolSize, catBoostOptions);
    AdjustPosteriorSamplingDeafultValues(trainDataMetaInfo, continueFromModel || continueFromProgress, catBoostOptions);
}