Medial Code Documentation
Loading...
Searching...
No Matches
Data Structures | Typedefs | Functions | Variables
xgboost::gbm Namespace Reference

Copyright 2019-2023, XGBoost Contributors. More...

Data Structures

class  Dart
 
struct  DartTrainParam
 training parameters More...
 
struct  DeprecatedGBLinearModelParam
 
class  GBLinear
 gradient boosted linear model More...
 
class  GBLinearModel
 
struct  GBLinearTrainParam
 
class  GBTree
 
struct  GBTreeModel
 
struct  GBTreeModelParam
 model parameters More...
 
struct  GBTreeTrainParam
 training parameters More...
 

Typedefs

using TreesOneGroup = std::vector< std::unique_ptr< RegTree > >
 Container for all trees built (not update) for one group.
 
using TreesOneIter = std::vector< TreesOneGroup >
 Container for all trees built (not update) for one iteration.
 

Functions

 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)
 

Variables

Context const * ctx
 
return p
 

Detailed Description

Copyright 2019-2023, XGBoost Contributors.

Function Documentation

◆ GPUDartPredictInc()

void xgboost::gbm::GPUDartPredictInc ( common::Span< float >  ,
common::Span< float >  ,
float  ,
size_t  ,
bst_group_t  ,
bst_group_t   
)

Increment the prediction on GPU.

Parameters
out_predtsPrediction for the whole model.
predtsPrediction for current tree.
tree_wTree weight.

Variable Documentation

◆ ctx

Context const * xgboost::gbm::ctx
Initial value:
{
return new GBLinear(booster_config, ctx)