Function ExtractBucketIndex

catboost/private/libs/algo/leafwise_scoring.cpp:100–136

template <class TColumn, typename TBucketIndexType>
inline static void ExtractBucketIndex(
    const TCalcScoreFold& fold,
    const TColumn& column,
    bool isEstimatedData,
    bool isOnlineData,
    TIndexRange<ui32> docIndexRange,
    int groupSize,
    const TVector<ui32>& groupPartsBucketsOffsets,
    TVector<TBucketIndexType>* bucketIdx // already of proper size
) {
    if (const auto* denseColumnData
        = dynamic_cast<const TCompressedValuesHolderImpl<TColumn>*>(&column))
    {
        const ui32* objectIndexing;
        int beginOffset;
        GetIndexingParams(fold, isEstimatedData, isOnlineData, &objectIndexing, &beginOffset);

        const TCompressedArray& compressedArray = *denseColumnData->GetCompressedData().GetSrc();

        compressedArray.DispatchBitsPerKeyToDataType(
            "ExtractBucketIndex",
            [&] (const auto* columnData) {
                SetBucketIndex(
                    columnData,
                    objectIndexing,
                    beginOffset,
                    docIndexRange,
                    groupSize,
                    groupPartsBucketsOffsets,
                    bucketIdx
                );
            }
        );
    } else {
        CB_ENSURE(false, "ExtractBucketIndex: unexpected column type");
    }
}


// Calculate index of leaf for each document given a new split ensemble.