objective.h

前往此文件文档。

 
#ifndef XGBOOST_OBJECTIVE_H_
#define XGBOOST_OBJECTIVE_H_
 
#include <dmlc/registry.h>
#include <xgboost/base.h>
#include <xgboost/data.h>
#include <xgboost/host_device_vector.h>
#include <xgboost/linalg.h> // for Vector
#include <xgboost/model.h>
#include <xgboost/task.h>
 
#include <cstdint> // for int32_t
#include <functional>
#include <string> // for string
 
namespace xgboost {
 
class RegTree;
struct Context;
 
class ObjFunction : public Configurable {
 protected
 Context const* ctx_{nullptr};
 
 public
 static constexpr float DefaultBaseScore() { return 0.5f; }
 
 public
 ~ObjFunction() override = default;
 virtual void Configure(Args const& args) = 0;
 virtual void GetGradient(HostDeviceVector<float> const& preds, MetaInfo const& info,
 std::int32_t iter, linalg::Matrix<GradientPair>* out_gpair) = 0;
 
 [[nodiscard]] virtual const char* DefaultEvalMetric() const = 0;
 [[nodiscard]] virtual Json DefaultMetricConfig() const { return Json{Null{}}; }
 virtual void PredTransform(HostDeviceVector<float>*) const {}
 virtual void EvalTransform(HostDeviceVector<float>* io_preds) { this->PredTransform(io_preds); }
 virtual void ProbToMargin(linalg::Vector<float>* /*base_score*/) const {}
 virtual void InitEstimation(MetaInfo const& info, linalg::Vector<float>* base_score) const;
 [[nodiscard]] virtual struct ObjInfo Task() const = 0;
 [[nodiscard]] virtual bst_target_t Targets(MetaInfo const& info) const {
 if (info.labels.Shape(1) > 1) {
 LOG(FATAL) << "当前目标函数不支持多输出";
    }
 return 1;
  }
 [[nodiscard]] Context const* Ctx() const { return this->ctx_; }
 
 virtual void UpdateTreeLeaf(HostDeviceVector<bst_node_t> const& /*position*/,
 MetaInfo const& /*info*/, float /*learning_rate*/,
 HostDeviceVector<float> const& /*prediction*/,
 std::int32_t /*group_idx*/, RegTree* /*p_tree*/) const {}
 static ObjFunction* Create(const std::string& name, Context const* ctx);
};
 
struct ObjFunctionReg
 : public dmlc::FunctionRegEntryBase<ObjFunctionReg,
 std::function<ObjFunction* ()> > {
};
 
#define XGBOOST_REGISTER_OBJECTIVE(UniqueId, Name) \
 static DMLC_ATTRIBUTE_UNUSED ::xgboost::ObjFunctionReg & \
 __make_ ## ObjFunctionReg ## _ ## UniqueId ## __ = \
 ::dmlc::Registry< ::xgboost::ObjFunctionReg>::Get()->__REGISTER__(Name)
} // namespace xgboost
#endif // XGBOOST_OBJECTIVE_H_