Function BestWeightedSplit

library/cpp/grid_creator/binarization.cpp:1647–1673

THashSet<float> BestWeightedSplit(
    TVector<float>&& featureValues,
    const TVector<float>& weights,
    int maxBordersCount,
    EBorderSelectionType borderSelectionType,
    bool filterNans,
    bool featuresAreSorted
) {
    switch (borderSelectionType) {
        case EBorderSelectionType::MinEntropy:
            return BestWeightedSplit<EPenaltyType::MinEntropy>(std::move(featureValues), weights, maxBordersCount,
                EOptimizationType::Exact, filterNans, featuresAreSorted);
        case EBorderSelectionType::MaxLogSum:
            return BestWeightedSplit<EPenaltyType::MaxSumLog>(std::move(featureValues), weights, maxBordersCount,
                EOptimizationType::Exact, filterNans, featuresAreSorted);
        case EBorderSelectionType ::GreedyLogSum:
            return BestWeightedSplit<EPenaltyType::MaxSumLog>(std::move(featureValues), weights, maxBordersCount,
                EOptimizationType::Greedy, filterNans, featuresAreSorted);
        case EBorderSelectionType ::GreedyMinEntropy:
            return BestWeightedSplit<EPenaltyType::MinEntropy>(std::move(featureValues), weights, maxBordersCount,
                EOptimizationType::Greedy, filterNans, featuresAreSorted);
        default:
            const auto borderSelectionTypeName = GetEnumNames<EBorderSelectionType>().at(borderSelectionType);
            throw (yexception() << "Weights are unsupported for " << borderSelectionTypeName <<
                                " border selection type.");
    }
}


template <EPenaltyType PenaltyType>