Medial Code Documentation
Loading...
Searching...
No Matches
Public Member Functions | Data Fields
xgboost::LearnerModelParam Struct Reference

Basic model parameters, used to describe the booster. More...

#include <learner.h>

Public Member Functions

 LearnerModelParam (Context const *ctx, LearnerModelParamLegacy const &user_param, linalg::Tensor< float, 1 > base_margin, ObjInfo t, MultiStrategy multi_strategy)
 
 LearnerModelParam (LearnerModelParamLegacy const &user_param, ObjInfo t, MultiStrategy multi_strategy)
 
 LearnerModelParam (bst_feature_t n_features, linalg::Tensor< float, 1 > base_score, std::uint32_t n_groups, bst_target_t n_targets, MultiStrategy multi_strategy)
 
linalg::TensorView< float const, 1 > BaseScore (Context const *ctx) const
 
linalg::TensorView< float const, 1 > BaseScore (std::int32_t device) const
 
void Copy (LearnerModelParam const &that)
 
bool IsVectorLeaf () const noexcept
 
bst_target_t OutputLength () const noexcept
 
bst_target_t LeafLength () const noexcept
 
bool Initialized () const
 

Data Fields

bst_feature_t num_feature {0}
 The number of features.
 
std::uint32_t num_output_group {0}
 The number of classes or targets.
 
ObjInfo task {ObjInfo::kRegression}
 Current task, determined by objective.
 
MultiStrategy multi_strategy {MultiStrategy::kOneOutputPerTree}
 Strategy for building multi-target models.
 

Detailed Description

Basic model parameters, used to describe the booster.


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