xgboost
|
Basic model parameters, used to describe the booster. More...
#include <learner.h>
Public Member Functions | |
LearnerModelParam ()=default | |
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 (DeviceOrd 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 |
Public Attributes | |
bst_feature_t | num_feature {0} |
The number of features. More... | |
std::uint32_t | num_output_group {0} |
The number of classes or targets. More... | |
ObjInfo | task {ObjInfo::kRegression} |
Current task, determined by objective. More... | |
MultiStrategy | multi_strategy {MultiStrategy::kOneOutputPerTree} |
Strategy for building multi-target models. More... | |
Basic model parameters, used to describe the booster.
|
default |
xgboost::LearnerModelParam::LearnerModelParam | ( | Context const * | ctx, |
LearnerModelParamLegacy const & | user_param, | ||
linalg::Tensor< float, 1 > | base_margin, | ||
ObjInfo | t, | ||
MultiStrategy | multi_strategy | ||
) |
xgboost::LearnerModelParam::LearnerModelParam | ( | LearnerModelParamLegacy const & | user_param, |
ObjInfo | t, | ||
MultiStrategy | multi_strategy | ||
) |
|
inline |
linalg::TensorView<float const, 1> xgboost::LearnerModelParam::BaseScore | ( | Context const * | ctx | ) | const |
linalg::TensorView<float const, 1> xgboost::LearnerModelParam::BaseScore | ( | DeviceOrd | device | ) | const |
void xgboost::LearnerModelParam::Copy | ( | LearnerModelParam const & | that | ) |
|
inline |
|
inlinenoexcept |
|
inlinenoexcept |
|
inlinenoexcept |
MultiStrategy xgboost::LearnerModelParam::multi_strategy {MultiStrategy::kOneOutputPerTree} |
Strategy for building multi-target models.
bst_feature_t xgboost::LearnerModelParam::num_feature {0} |
The number of features.
std::uint32_t xgboost::LearnerModelParam::num_output_group {0} |
The number of classes or targets.
ObjInfo xgboost::LearnerModelParam::task {ObjInfo::kRegression} |
Current task, determined by objective.