xgboost
Public Member Functions | Public Attributes | List of all members
xgboost::LearnerModelParam Struct Reference

#include <learner.h>

Collaboration diagram for xgboost::LearnerModelParam:
Collaboration graph

Public Member Functions

 LearnerModelParam ()=default
 
 LearnerModelParam (Context const *ctx, LearnerModelParamLegacy const &user_param, linalg::Tensor< float, 1 > base_margin, ObjInfo t)
 
 LearnerModelParam (LearnerModelParamLegacy const &user_param, ObjInfo t)
 
 LearnerModelParam (bst_feature_t n_features, linalg::Tensor< float, 1 > base_margin, uint32_t n_groups)
 
linalg::TensorView< float const, 1 > BaseScore (Context const *ctx) const
 
linalg::TensorView< float const, 1 > BaseScore (int32_t device) const
 
void Copy (LearnerModelParam const &that)
 
bool Initialized () const
 

Public Attributes

uint32_t num_feature { 0 }
 
uint32_t num_output_group { 0 }
 
ObjInfo task {ObjInfo::kRegression}
 

Constructor & Destructor Documentation

◆ LearnerModelParam() [1/4]

xgboost::LearnerModelParam::LearnerModelParam ( )
default

◆ LearnerModelParam() [2/4]

xgboost::LearnerModelParam::LearnerModelParam ( Context const *  ctx,
LearnerModelParamLegacy const &  user_param,
linalg::Tensor< float, 1 >  base_margin,
ObjInfo  t 
)

◆ LearnerModelParam() [3/4]

xgboost::LearnerModelParam::LearnerModelParam ( LearnerModelParamLegacy const &  user_param,
ObjInfo  t 
)

◆ LearnerModelParam() [4/4]

xgboost::LearnerModelParam::LearnerModelParam ( bst_feature_t  n_features,
linalg::Tensor< float, 1 >  base_margin,
uint32_t  n_groups 
)
inline

Member Function Documentation

◆ BaseScore() [1/2]

linalg::TensorView<float const, 1> xgboost::LearnerModelParam::BaseScore ( Context const *  ctx) const

◆ BaseScore() [2/2]

linalg::TensorView<float const, 1> xgboost::LearnerModelParam::BaseScore ( int32_t  device) const

◆ Copy()

void xgboost::LearnerModelParam::Copy ( LearnerModelParam const &  that)

◆ Initialized()

bool xgboost::LearnerModelParam::Initialized ( ) const
inline

Member Data Documentation

◆ num_feature

uint32_t xgboost::LearnerModelParam::num_feature { 0 }

◆ num_output_group

uint32_t xgboost::LearnerModelParam::num_output_group { 0 }

◆ task

ObjInfo xgboost::LearnerModelParam::task {ObjInfo::kRegression}

The documentation for this struct was generated from the following file: