|
Medial Code Documentation
|
Copyright 2014-2023 by XGBoost Contributors. More...
#include <algorithm>#include <cmath>#include <cstring>#include <limits>#include <string>#include <vector>#include "../common/categorical.h"#include "../common/linalg_op.h"#include "../common/math.h"#include "xgboost/data.h"#include "xgboost/linalg.h"#include "xgboost/parameter.h"Go to the source code of this file.
Data Structures | |
| struct | xgboost::tree::TrainParam |
| training parameters for regression tree More... | |
| struct | xgboost::tree::SplitEntryContainer< GradientT > |
| statistics that is helpful to store and represent a split solution for the tree More... | |
Namespaces | |
| namespace | xgboost |
| namespace of xgboost | |
| namespace | xgboost::tree |
| Copyright 2021-2023 by XGBoost Contributors. | |
| namespace | std |
Typedefs | |
| using | xgboost::tree::SplitEntry = SplitEntryContainer< GradStats > |
Functions | |
| template<typename TrainingParams , typename T > | |
| XGBOOST_DEVICE T | xgboost::tree::CalcGainGivenWeight (const TrainingParams &p, T sum_grad, T sum_hess, T w) |
| template<typename TrainingParams , typename T > | |
| XGBOOST_DEVICE T | xgboost::tree::CalcWeight (const TrainingParams &p, T sum_grad, T sum_hess) |
| template<typename TrainingParams , typename T > | |
| XGBOOST_DEVICE T | xgboost::tree::CalcGain (const TrainingParams &p, T sum_grad, T sum_hess) |
| template<typename TrainingParams , typename StatT , typename T = decltype(StatT().GetHess())> | |
| XGBOOST_DEVICE T | xgboost::tree::CalcGain (const TrainingParams &p, StatT stat) |
| template<typename TrainingParams , typename GpairT > | |
| XGBOOST_DEVICE float | xgboost::tree::CalcWeight (const TrainingParams &p, GpairT sum_grad) |
| void | xgboost::tree::CalcWeight (TrainParam const &p, linalg::VectorView< GradientPairPrecise const > grad_sum, float eta, linalg::VectorView< float > out_w) |
| multi-target weight, calculated with learning rate. | |
| void | xgboost::tree::CalcWeight (TrainParam const &p, linalg::VectorView< GradientPairPrecise const > grad_sum, linalg::VectorView< float > out_w) |
| multi-target weight | |
| double | xgboost::tree::CalcGainGivenWeight (TrainParam const &p, linalg::VectorView< GradientPairPrecise const > sum_grad, linalg::VectorView< float const > weight) |
| struct | xgboost::tree::XGBOOST_ALIGNAS (16) GradStats |
| core statistics used for tree construction | |
| template<typename T , typename U > | |
| std::vector< T > & | xgboost::tree::CopyStats (linalg::VectorView< U > const &src, std::vector< T > *dst) |
| GradStats & | xgboost::tree::CopyStats (GradStats const &src, GradStats *dst) |
| void | xgboost::ParseInteractionConstraint (std::string const &constraint_str, std::vector< std::vector< xgboost::bst_feature_t > > *p_out) |
| std::ostream & | std::operator<< (std::ostream &os, const std::vector< int > &t) |
| std::istream & | std::operator>> (std::istream &is, std::vector< int > &t) |
Copyright 2014-2023 by XGBoost Contributors.
training parameters, statistics used to support tree construction.