Medial Code Documentation
|
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. | |
Basic model parameters, used to describe the booster.