linear_updater.h

转到此文件的文档。

/*
 * 由 Contributors 2018 版权所有
 */
#pragma once

#include <dmlc/registry.h>
#include <xgboost/base.h>
#include <xgboost/data.h>
#include <xgboost/host_device_vector.h>
#include <xgboost/model.h>

#include <functional>
#include <string>
#include <utility>
#include <vector>


namespace xgboost {

class Json;
struct Context;

namespace gbm {
class GBLinearModel;
} // 命名空间 gbm

/** \brief interface of linear updater */class LinearUpdater : public Configurable {
 protected
 Context const* ctx_;

 public
 /** \brief virtual destructor */~LinearUpdater() override = default;
 /**
  * \brief Initialize the updater with given arguments.
  */
 /**
  * \brief Updates linear model given gradients.
  */
 /**
  * \brief Create a linear updater given name.
  */
/** \brief Registry entry for linear updater. */struct LinearUpdaterReg
 : public dmlc::FunctionRegEntryBase<LinearUpdaterReg,
 std::function<LinearUpdater*()> > {};

#define XGBOOST_REGISTER_LINEAR_UPDATER(UniqueId, Name) \
 static DMLC_ATTRIBUTE_UNUSED ::xgboost::LinearUpdaterReg& \
 __make_##LinearUpdaterReg##_##UniqueId##__ = \
 ::dmlc::Registry< ::xgboost::LinearUpdaterReg>::Get()->__REGISTER__( \
 Name)

} // 命名空间 xgboost