xgboost
Static Public Member Functions | List of all members
xgboost::common::AFTLoss< Distribution > Struct Template Reference

The AFT loss function. More...

#include <survival_util.h>

Collaboration diagram for xgboost::common::AFTLoss< Distribution >:
Collaboration graph

Static Public Member Functions

static XGBOOST_DEVICE double Loss (double y_lower, double y_upper, double y_pred, double sigma)
 
static XGBOOST_DEVICE double Gradient (double y_lower, double y_upper, double y_pred, double sigma)
 
static XGBOOST_DEVICE double Hessian (double y_lower, double y_upper, double y_pred, double sigma)
 

Detailed Description

template<typename Distribution>
struct xgboost::common::AFTLoss< Distribution >

The AFT loss function.

Member Function Documentation

◆ Gradient()

template<typename Distribution >
static XGBOOST_DEVICE double xgboost::common::AFTLoss< Distribution >::Gradient ( double  y_lower,
double  y_upper,
double  y_pred,
double  sigma 
)
inlinestatic

◆ Hessian()

template<typename Distribution >
static XGBOOST_DEVICE double xgboost::common::AFTLoss< Distribution >::Hessian ( double  y_lower,
double  y_upper,
double  y_pred,
double  sigma 
)
inlinestatic

◆ Loss()

template<typename Distribution >
static XGBOOST_DEVICE double xgboost::common::AFTLoss< Distribution >::Loss ( double  y_lower,
double  y_upper,
double  y_pred,
double  sigma 
)
inlinestatic

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