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.