Medial Code Documentation
|
Common cached items for ranking tasks. More...
#include <ranking_utils.h>
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< GradientPair > | CUDARounding (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 |
Common cached items for ranking tasks.