Medial Code Documentation
Loading...
Searching...
No Matches
Public Member Functions | Protected Member Functions
xgboost::ltr::RankingCache Class Reference

Common cached items for ranking tasks. More...

#include <ranking_utils.h>

Inheritance diagram for xgboost::ltr::RankingCache:
xgboost::ltr::MAPCache xgboost::ltr::NDCGCache xgboost::ltr::PreCache

Public Member Functions

 RankingCache (Context const *ctx, MetaInfo const &info, LambdaRankParam const &p)
 
std::size_t MaxPositionSize () const
 
common::Span< bst_group_t const > DataGroupPtr (Context const *ctx) const
 
auto const & Param () const
 
std::size_t Groups () const
 
double WeightNorm () const
 
common::Span< std::size_t const > SortedIdx (Context const *ctx, common::Span< float const > predt)
 
common::Span< std::size_t > SortedIdxY (Context const *ctx, std::size_t n_samples)
 
common::Span< float > RankedY (Context const *ctx, std::size_t n_samples)
 
common::Span< std::size_t const > CUDAThreadsGroupPtr () const
 
std::size_t CUDAThreads () const
 
linalg::VectorView< GradientPairCUDARounding (Context const *ctx)
 
common::Span< double > CUDACostRounding (Context const *ctx)
 
template<typename Type >
common::Span< Type > MaxLambdas (Context const *ctx, std::size_t n)
 

Protected Member Functions

std::size_t MaxGroupSize () const
 

Detailed Description

Common cached items for ranking tasks.


The documentation for this class was generated from the following files: