Medial Code Documentation
|
Public Member Functions | |
template<typename GradientSumT > | |
XGBOOST_DEVICE float | CalcSplitGain (const ParamT ¶m, 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 ¶m, GradientSumT const &stats) const |
template<typename GradientSumT > | |
XGBOOST_DEVICE float | CalcWeightCat (ParamT const ¶m, 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 |