// Copyright by Contributors #include #include "../helpers.h" #include "xgboost/gbm.h" typedef std::pair arg; TEST(Linear, shotgun) { typedef std::pair arg; auto mat = xgboost::CreateDMatrix(10, 10, 0); auto updater = std::unique_ptr( xgboost::LinearUpdater::Create("shotgun")); updater->Init({{"eta", "1."}}); xgboost::HostDeviceVector gpair( (*mat)->Info().num_row_, xgboost::GradientPair(-5, 1.0)); xgboost::gbm::GBLinearModel model; model.param.num_feature = (*mat)->Info().num_col_; model.param.num_output_group = 1; model.LazyInitModel(); updater->Update(&gpair, (*mat).get(), &model, gpair.Size()); ASSERT_EQ(model.bias()[0], 5.0f); delete mat; } TEST(Linear, coordinate) { typedef std::pair arg; auto mat = xgboost::CreateDMatrix(10, 10, 0); auto updater = std::unique_ptr( xgboost::LinearUpdater::Create("coord_descent")); updater->Init({{"eta", "1."}}); xgboost::HostDeviceVector gpair( (*mat)->Info().num_row_, xgboost::GradientPair(-5, 1.0)); xgboost::gbm::GBLinearModel model; model.param.num_feature = (*mat)->Info().num_col_; model.param.num_output_group = 1; model.LazyInitModel(); updater->Update(&gpair, (*mat).get(), &model, gpair.Size()); ASSERT_EQ(model.bias()[0], 5.0f); delete mat; }