|
| DMLC_REGISTRY_FILE_TAG (gblinear) |
|
void | LinearCheckLayer (unsigned layer_begin) |
|
| DMLC_REGISTER_PARAMETER (GBLinearTrainParam) |
|
| describe ("Linear booster, implement generalized linear model.") .set_body([](LearnerModelParam const *booster_config |
|
| DMLC_REGISTER_PARAMETER (DeprecatedGBLinearModelParam) |
|
| DMLC_REGISTRY_LINK_TAG (gblinear) |
|
| DMLC_REGISTRY_LINK_TAG (gbtree) |
|
| DMLC_REGISTRY_FILE_TAG (gbtree) |
|
void | GPUCopyGradient (HostDeviceVector< GradientPair > const *, bst_group_t, bst_group_t, HostDeviceVector< GradientPair > *) |
|
void | CopyGradient (HostDeviceVector< GradientPair > const *in_gpair, int32_t n_threads, bst_group_t n_groups, bst_group_t group_id, HostDeviceVector< GradientPair > *out_gpair) |
|
void | GPUDartPredictInc (common::Span< float >, common::Span< float >, float, size_t, bst_group_t, bst_group_t) |
| Increment the prediction on GPU.
|
|
void | GPUDartInplacePredictInc (common::Span< float >, common::Span< float >, float, size_t, linalg::TensorView< float const, 1 >, bst_group_t, bst_group_t) |
|
| DMLC_REGISTER_PARAMETER (GBTreeModelParam) |
|
| DMLC_REGISTER_PARAMETER (GBTreeTrainParam) |
|
| DMLC_REGISTER_PARAMETER (DartTrainParam) |
|
| describe ("Tree booster, gradient boosted trees.") .set_body([](LearnerModelParam const *booster_config |
|
| describe ("Tree booster, dart.") .set_body([](LearnerModelParam const *booster_config |
|
| TEST (GBLinear, JsonIO) |
|
Copyright 2019-2023, XGBoost Contributors.