task.h

Go to the documentation of this file.

 
#ifndef XGBOOST_TASK_H_
#define XGBOOST_TASK_H_
 
#include <xgboost/base.h>
 
#include <cinttypes>
 
namespace xgboost {
struct ObjInfo {
  // What kind of problem are we trying to solve
  enum Task : 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};
 
  explicit ObjInfo(Task t) : task{t} {}
  ObjInfo(Task t, bool khess) : task{t}, const_hess{khess} {}
};
}  // namespace xgboost
#endif  // XGBOOST_TASK_H_