Function CalcMemoryForFindBestSplit

library/cpp/grid_creator/binarization.cpp:1824–1853

namespace NSplitSelection {

    size_t CalcMemoryForFindBestSplit(
        int maxBordersCount,
        size_t nonDefaultObjectCount,
        const TMaybe<TDefaultValue<float>>& defaultValue,
        EBorderSelectionType type) {

        switch (type) {
            case EBorderSelectionType::Median:
            case EBorderSelectionType::UniformAndQuantiles:
                return maxBordersCount * sizeof(float)
                    + (defaultValue ? ((nonDefaultObjectCount + 1) * sizeof(float)) : 0);
            case EBorderSelectionType::Uniform:
                return maxBordersCount * sizeof(float);
            case EBorderSelectionType::GreedyLogSum:
            case EBorderSelectionType::GreedyMinEntropy:
                return CalcMemoryForFindBestSplitGreedyBinarizer(
                    maxBordersCount,
                    nonDefaultObjectCount,
                    defaultValue);

            case EBorderSelectionType::MinEntropy:
            case EBorderSelectionType::MaxLogSum:
                return CalcMemoryForFindBestSplitExactBinarizer(
                    maxBordersCount,
                    nonDefaultObjectCount,
                    defaultValue);
            default:
                Y_UNREACHABLE();
        }
    }

}