Parameter structure for AFT loss and metric. More...
#include <survival_util.h>
Inheritance diagram for xgboost::common::AFTParam:
Collaboration diagram for xgboost::common::AFTParam:
Public Member Functions | |
| DMLC_DECLARE_PARAMETER (AFTParam) | |
 Public Member Functions inherited from xgboost::XGBoostParameter< AFTParam > | |
| Args | UpdateAllowUnknown (Container const &kwargs, bool *out_changed=nullptr) |
| bool | GetInitialised () const |
Public Attributes | |
| ProbabilityDistributionType | aft_loss_distribution |
| Choice of probability distribution for the noise term in AFT. More... | |
| float | aft_loss_distribution_scale |
| Scaling factor to be applied to the distribution. More... | |
Additional Inherited Members | |
| Protected Attributes inherited from xgboost::XGBoostParameter< AFTParam > | |
| bool | initialised_ |
Detailed Description
Parameter structure for AFT loss and metric.
Member Function Documentation
◆ DMLC_DECLARE_PARAMETER()
|
inline |
Member Data Documentation
◆ aft_loss_distribution
| ProbabilityDistributionType xgboost::common::AFTParam::aft_loss_distribution |
Choice of probability distribution for the noise term in AFT.
◆ aft_loss_distribution_scale
| float xgboost::common::AFTParam::aft_loss_distribution_scale |
Scaling factor to be applied to the distribution.
The documentation for this struct was generated from the following file:
- /home/runner/work/xgboost/xgboost/src/common/survival_util.h
