706be4e5d4
* fix rebase conflict * [core] additional gblinear improvements * [R] callback for gblinear coefficients history * force eta=1 for gblinear python tests * add top_k to GreedyFeatureSelector * set eta=1 in shotgun test * [core] fix SparsePage processing in gblinear; col-wise multithreading in greedy updater * set sorted flag within TryInitColData * gblinear tests: use scale, add external memory test * fix multiclass for greedy updater * fix whitespace * fix typo
44 lines
1.6 KiB
C++
44 lines
1.6 KiB
C++
// Copyright by Contributors
|
|
#include <xgboost/linear_updater.h>
|
|
#include "../helpers.h"
|
|
#include "xgboost/gbm.h"
|
|
|
|
typedef std::pair<std::string, std::string> arg;
|
|
|
|
TEST(Linear, shotgun) {
|
|
typedef std::pair<std::string, std::string> arg;
|
|
auto mat = CreateDMatrix(10, 10, 0);
|
|
std::vector<bool> enabled(mat->info().num_col, true);
|
|
mat->InitColAccess(enabled, 1.0f, 1 << 16, false);
|
|
auto updater = std::unique_ptr<xgboost::LinearUpdater>(
|
|
xgboost::LinearUpdater::Create("shotgun"));
|
|
updater->Init({{"eta", "1."}});
|
|
std::vector<xgboost::bst_gpair> gpair(mat->info().num_row,
|
|
xgboost::bst_gpair(-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);
|
|
}
|
|
|
|
TEST(Linear, coordinate) {
|
|
typedef std::pair<std::string, std::string> arg;
|
|
auto mat = CreateDMatrix(10, 10, 0);
|
|
std::vector<bool> enabled(mat->info().num_col, true);
|
|
mat->InitColAccess(enabled, 1.0f, 1 << 16, false);
|
|
auto updater = std::unique_ptr<xgboost::LinearUpdater>(
|
|
xgboost::LinearUpdater::Create("coord_descent"));
|
|
updater->Init({});
|
|
std::vector<xgboost::bst_gpair> gpair(mat->info().num_row,
|
|
xgboost::bst_gpair(-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);
|
|
} |