Medial Code Documentation
Loading...
Searching...
No Matches
Public Member Functions
xgboost::ltr::NDCGCache Class Reference
Inheritance diagram for xgboost::ltr::NDCGCache:
xgboost::ltr::RankingCache

Public Member Functions

void InitOnCPU (Context const *ctx, MetaInfo const &info)
 
void InitOnCUDA (Context const *ctx, MetaInfo const &info)
 
 NDCGCache (Context const *ctx, MetaInfo const &info, LambdaRankParam const &p)
 
linalg::VectorView< double const > InvIDCG (Context const *ctx) const
 
common::Span< double const > Discount (Context const *ctx) const
 
linalg::VectorView< double > Dcg (Context const *ctx)
 
- Public Member Functions inherited from xgboost::ltr::RankingCache
 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)
 

Additional Inherited Members

- Protected Member Functions inherited from xgboost::ltr::RankingCache
std::size_t MaxGroupSize () const
 

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