Config for linear updaters. (#5222)
This commit is contained in:
@@ -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_;
|
||||
|
||||
|
||||
Reference in New Issue
Block a user