Medial Code Documentation
Loading...
Searching...
No Matches
Public Member Functions | Data Fields
xgboost::tree::TreeEvaluator::SplitEvaluator< ParamT > Struct Template Reference

Public Member Functions

template<typename GradientSumT >
XGBOOST_DEVICE float CalcSplitGain (const ParamT &param, bst_node_t nidx, bst_feature_t fidx, GradientSumT const &left, GradientSumT const &right) const
 
template<typename GradientSumT >
XGBOOST_DEVICE float CalcWeight (bst_node_t nodeid, const ParamT &param, GradientSumT const &stats) const
 
template<typename GradientSumT >
XGBOOST_DEVICE float CalcWeightCat (ParamT const &param, GradientSumT const &stats) const
 
XGBOOST_DEVICE float Divide (float a, float b) const
 
template<typename GradientSumT >
XGBOOST_DEVICE float CalcGainGivenWeight (ParamT const &p, GradientSumT const &stats, float w) const
 
template<typename GradientSumT >
XGBOOST_DEVICE float CalcGain (bst_node_t nid, ParamT const &p, GradientSumT const &stats) const
 

Data Fields

const int * constraints
 
const float * lower
 
const float * upper
 
bool has_constraint
 

The documentation for this struct was generated from the following file: