Method Split

catboost/cuda/gpu_data/ut/test_bin_builder.cpp:111–161

        }

        void Split(const TBinarySplit& split,
                   ui32 depth,
                   TVector<ui32>& bins) {
            if (FeaturesManager.IsFloat(split.FeatureId)) {
                const auto floatFeatureIdx = DataProvider.MetaInfo.FeaturesLayout->GetInternalFeatureIdx<EFeatureType::Float>(split.FeatureId);
                auto& valuesHolder = **(DataProvider.ObjectsData->GetFloatFeature(*floatFeatureIdx));
                auto featureBins = valuesHolder.ExtractValues<ui16>(&NPar::LocalExecutor());
                for (ui32 i = 0; i < bins.size(); ++i) {
                    bins[i] |= (featureBins[i] > split.BinIdx) << depth;
                }
            } else if (FeaturesManager.IsCat(split.FeatureId)) {
                const auto catFeatureIdx = DataProvider.MetaInfo.FeaturesLayout->GetInternalFeatureIdx<EFeatureType::Categorical>(split.FeatureId);
                auto& valuesHolder = **(DataProvider.ObjectsData->GetCatFeature(*catFeatureIdx));
                auto featureBins = valuesHolder.ExtractValues<ui32>(&NPar::LocalExecutor());
                for (ui32 i = 0; i < bins.size(); ++i) {
                    bins[i] |= (featureBins[i] == split.BinIdx) << depth;
                }
            } else {
                const auto& ctr = FeaturesManager.GetCtr(split.FeatureId);
                auto treeSplit = BuildTreeCtrSplitCpu(ctr.FeatureTensor);

                const auto& borders = FeaturesManager.GetBorders(split.FeatureId);

                TCpuTargetClassCtrCalcer calcer(treeSplit.UniqueCount,
                                                treeSplit.Bins,
                                                GetWeights(*DataProvider.TargetData),
                                                ctr.Configuration.Prior[0], ctr.Configuration.Prior[1]);

                TVector<ui32> featureBins;
                if (ctr.Configuration.Type == ECtrType::FeatureFreq) {
                    auto freqCtr = calcer.ComputeFreqCtr();
                    featureBins = NCB::BinarizeLine<ui32>(freqCtr, ENanMode::Forbidden, borders);
                } else if (ctr.Configuration.Type == ECtrType::Buckets) {
                    auto floatCtr = calcer.Calc(Indices, TConstArrayRef<ui32>(Indices), BinarizedTarget, NumClasses);
                    TVector<float> values;
                    for (ui32 i = 0; i < treeSplit.Bins.size(); ++i) {
                        values.push_back(floatCtr[i][ctr.Configuration.ParamId]);
                    }
                    featureBins = NCB::BinarizeLine<ui32>(values,
                                                          ENanMode::Forbidden,
                                                          borders);
                } else {
                    ythrow yexception() << "Test for ctr type " << ctr.Configuration.Type
                                        << " isn't supported currently " << Endl;
                }

                for (ui32 i = 0; i < bins.size(); ++i) {
                    bins[i] |= (featureBins[i] > split.BinIdx) << depth;
                }
            }
        }
    };

    void CheckBins(const TFeatureParallelDataSet& dataSet,