task.h

前往此文件的文档。

#ifndef XGBOOST_TASK_H_
#define XGBOOST_TASK_H_

#include <xgboost/base.h>

#include <cstdint> // for uint8_t

namespace xgboost {
struct ObjInfo {
 // 我们试图解决什么类型的问题
 enum Task : std::uint8_t {
 kRegression = 0,
 kBinary = 1,
 kClassification = 2,
 kSurvival = 3,
 kRanking = 4,
 kOther = 5,
 } task;
 // 目标函数是否具有常数Hessian值？
 bool const_hess{false};
 bool zero_hess{false};

 ObjInfo(Task t) : task{t} {} // NOLINT
 ObjInfo(Task t, bool khess, bool zhess) : task{t}, const_hess{khess}, zero_hess(zhess) {}

 XGBOOST_DEVICE bool UpdateTreeLeaf() const { return zero_hess; }
};
} // namespace xgboost
#endif // XGBOOST_TASK_H_