Medial Code Documentation
Loading...
Searching...
No Matches
Data Structures | Namespaces | Functions | Variables
gbtree.cc File Reference

Copyright 2014-2023 by Contributors. More...

#include "gbtree.h"
#include <dmlc/omp.h>
#include <dmlc/parameter.h>
#include <algorithm>
#include <cinttypes>
#include <limits>
#include <memory>
#include <string>
#include <utility>
#include <vector>
#include "../common/common.h"
#include "../common/error_msg.h"
#include "../common/random.h"
#include "../common/threading_utils.h"
#include "../common/timer.h"
#include "../data/proxy_dmatrix.h"
#include "gbtree_model.h"
#include "xgboost/base.h"
#include "xgboost/data.h"
#include "xgboost/gbm.h"
#include "xgboost/host_device_vector.h"
#include "xgboost/json.h"
#include "xgboost/logging.h"
#include "xgboost/model.h"
#include "xgboost/objective.h"
#include "xgboost/predictor.h"
#include "xgboost/string_view.h"
#include "xgboost/tree_model.h"
#include "xgboost/tree_updater.h"

Data Structures

class  xgboost::gbm::Dart
 

Namespaces

namespace  xgboost
 namespace of xgboost
 
namespace  xgboost::gbm
 Copyright 2019-2023, XGBoost Contributors.
 

Functions

 xgboost::gbm::DMLC_REGISTRY_FILE_TAG (gbtree)
 
void xgboost::gbm::GPUCopyGradient (HostDeviceVector< GradientPair > const *, bst_group_t, bst_group_t, HostDeviceVector< GradientPair > *)
 
void xgboost::gbm::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 xgboost::gbm::GPUDartPredictInc (common::Span< float >, common::Span< float >, float, size_t, bst_group_t, bst_group_t)
 Increment the prediction on GPU.
 
void xgboost::gbm::GPUDartInplacePredictInc (common::Span< float >, common::Span< float >, float, size_t, linalg::TensorView< float const, 1 >, bst_group_t, bst_group_t)
 
 xgboost::gbm::DMLC_REGISTER_PARAMETER (GBTreeModelParam)
 
 xgboost::gbm::DMLC_REGISTER_PARAMETER (GBTreeTrainParam)
 
 xgboost::gbm::DMLC_REGISTER_PARAMETER (DartTrainParam)
 
 xgboost::gbm::describe ("Tree booster, gradient boosted trees.") .set_body([](LearnerModelParam const *booster_config
 
 xgboost::gbm::describe ("Tree booster, dart.") .set_body([](LearnerModelParam const *booster_config
 

Variables

return xgboost::gbm::p
 

Detailed Description

Copyright 2014-2023 by Contributors.

gradient boosted tree implementation.

Author
Tianqi Chen