Prevent multiclass Hessian approaching 0 (#3304)

* Prevent Hessian in multiclass objective becoming zero

* Set default learning rate to 0.5 for "coord_descent" linear updater
This commit is contained in:
Rory Mitchell
2018-05-09 20:25:51 +12:00
committed by GitHub
parent b8a0d66fe6
commit 088bb4b27c
3 changed files with 4 additions and 3 deletions

View File

@@ -31,7 +31,7 @@ TEST(Linear, coordinate) {
mat->InitColAccess(enabled, 1.0f, 1 << 16, false);
auto updater = std::unique_ptr<xgboost::LinearUpdater>(
xgboost::LinearUpdater::Create("coord_descent"));
updater->Init({});
updater->Init({{"eta", "1."}});
xgboost::HostDeviceVector<xgboost::GradientPair> gpair(
mat->Info().num_row_, xgboost::GradientPair(-5, 1.0));
xgboost::gbm::GBLinearModel model;