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;
 // 目标函数是否有恒定的海森值？
 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_