xgboost
src
common
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
Generated on Mon Aug 22 2022 08:50:39 for xgboost by
1.8.17