task.h

Go to the documentation of this file.

 
#ifndef XGBOOST_TASK_H_
#define XGBOOST_TASK_H_
 
#include <xgboost/base.h>
 
#include <cstdint>  // for uint8_t
 
namespace xgboost {
struct ObjInfo {
  // What kind of problem are we trying to solve
  enum Task : std::uint8_t {
    kRegression = 0,
    kBinary = 1,
    kClassification = 2,
    kSurvival = 3,
    kRanking = 4,
    kOther = 5,
  } task;
  // Does the objective have constant hessian value?
  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_