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

@@ -5,12 +5,13 @@
#include <xgboost/gbm.h>
#include "../helpers.h"
#include "test_json_io.h"
#include "../../../src/gbm/gblinear_model.h"
#include "xgboost/base.h"
namespace xgboost {
TEST(Linear, shotgun) {
TEST(Linear, Shotgun) {
size_t constexpr kRows = 10;
size_t constexpr kCols = 10;
@@ -45,6 +46,10 @@ TEST(Linear, shotgun) {
delete pp_dmat;
}
TEST(Shotgun, JsonIO) {
TestUpdaterJsonIO("shotgun");
}
TEST(Linear, coordinate) {
size_t constexpr kRows = 10;
size_t constexpr kCols = 10;
@@ -72,4 +77,8 @@ TEST(Linear, coordinate) {
delete pp_dmat;
}
TEST(Coordinate, JsonIO){
TestUpdaterJsonIO("coord_descent");
}
} // namespace xgboost