xgboost

pseudo_huber.h
Go to the documentation of this file.
1 #ifndef XGBOOST_COMMON_PSEUDO_HUBER_H_
2 #define XGBOOST_COMMON_PSEUDO_HUBER_H_
3
6 #include "xgboost/parameter.h"
7
8 namespace xgboost {
9 struct PesudoHuberParam : public XGBoostParameter<PesudoHuberParam> {
10 float huber_slope{1.0};
11
12 DMLC_DECLARE_PARAMETER(PesudoHuberParam) {
13 DMLC_DECLARE_FIELD(huber_slope)
14 .set_default(1.0f)
15 .describe("The delta term in Pseudo-Huber loss.");
16 }
17 };
18 } // namespace xgboost
19 #endif // XGBOOST_COMMON_PSEUDO_HUBER_H_
parameter.h
macro for using C++11 enum class as DMLC parameter
xgboost::XGBoostParameter
Definition: parameter.h:84
xgboost::PesudoHuberParam::huber_slope
float huber_slope
Definition: pseudo_huber.h:10
xgboost::PesudoHuberParam
Definition: pseudo_huber.h:9
xgboost::PesudoHuberParam::DMLC_DECLARE_PARAMETER
DMLC_DECLARE_PARAMETER(PesudoHuberParam)
Definition: pseudo_huber.h:12
xgboost
namespace of xgboost
Definition: base.h:110