Config for linear updaters. (#5222)

This commit is contained in:
Jiaming Yuan
2020-01-25 11:26:46 +08:00
committed by GitHub
parent 40680368cf
commit 3eb1279bbf
14 changed files with 110 additions and 11 deletions

View File

@@ -8,6 +8,7 @@
#include <xgboost/data.h>
#include <xgboost/generic_parameters.h>
#include <xgboost/host_device_vector.h>
#include <xgboost/model.h>
#include <functional>
#include <string>
@@ -17,6 +18,8 @@
namespace xgboost {
class Json;
namespace gbm {
class GBLinearModel;
} // namespace gbm
@@ -24,7 +27,7 @@ class GBLinearModel;
/*!
* \brief interface of linear updater
*/
class LinearUpdater {
class LinearUpdater : public Configurable {
protected:
GenericParameter const* learner_param_;