Medial Code Documentation
Loading...
Searching...
No Matches
pseudo_huber.h
1#ifndef XGBOOST_COMMON_PSEUDO_HUBER_H_
2#define XGBOOST_COMMON_PSEUDO_HUBER_H_
6#include "xgboost/parameter.h"
7
8namespace xgboost {
9struct 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_
macro for using C++11 enum class as DMLC parameter
namespace of xgboost
Definition base.h:90
Definition pseudo_huber.h:9
Definition parameter.h:84