Medial Code Documentation
Loading...
Searching...
No Matches
Data Structures | Namespaces | Typedefs | Functions
param.h File Reference

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_DEVICExgboost::tree::CalcGainGivenWeight (const TrainingParams &p, T sum_grad, T sum_hess, T w)
 
template<typename TrainingParams , typename T >
XGBOOST_DEVICExgboost::tree::CalcWeight (const TrainingParams &p, T sum_grad, T sum_hess)
 
template<typename TrainingParams , typename T >
XGBOOST_DEVICExgboost::tree::CalcGain (const TrainingParams &p, T sum_grad, T sum_hess)
 
template<typename TrainingParams , typename StatT , typename T = decltype(StatT().GetHess())>
XGBOOST_DEVICExgboost::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)
 

Detailed Description

Copyright 2014-2023 by XGBoost Contributors.

training parameters, statistics used to support tree construction.

Author
Tianqi Chen