ad68865d6b
* Clean up logic for converting tree_method to updater sequence * Use C++11 enum class for extra safety Compiler will give warnings if switch statements don't handle all possible values of C++11 enum class. Also allow enum class to be used as DMLC parameter. * Fix compiler error + lint * Address reviewer comment * Better docstring for DECLARE_FIELD_ENUM_CLASS * Fix lint * Add C++ test to see if tree_method is recognized * Fix clang-tidy error * Add test_learner.h to R package * Update comments * Fix lint error
59 lines
2.0 KiB
C++
59 lines
2.0 KiB
C++
// Copyright by Contributors
|
|
#include <gtest/gtest.h>
|
|
#include <vector>
|
|
#include "helpers.h"
|
|
#include "./test_learner.h"
|
|
#include "xgboost/learner.h"
|
|
|
|
namespace xgboost {
|
|
|
|
class LearnerTestHookAdapter {
|
|
public:
|
|
static inline std::string GetUpdaterSequence(const Learner* learner) {
|
|
const LearnerTestHook* hook = dynamic_cast<const LearnerTestHook*>(learner);
|
|
CHECK(hook) << "LearnerImpl did not inherit from LearnerTestHook";
|
|
return hook->GetUpdaterSequence();
|
|
}
|
|
};
|
|
|
|
TEST(learner, Test) {
|
|
typedef std::pair<std::string, std::string> arg;
|
|
auto args = {arg("tree_method", "exact")};
|
|
auto mat_ptr = CreateDMatrix(10, 10, 0);
|
|
std::vector<std::shared_ptr<xgboost::DMatrix>> mat = {*mat_ptr};
|
|
auto learner = std::unique_ptr<Learner>(Learner::Create(mat));
|
|
learner->Configure(args);
|
|
|
|
delete mat_ptr;
|
|
}
|
|
|
|
TEST(learner, SelectTreeMethod) {
|
|
using arg = std::pair<std::string, std::string>;
|
|
auto mat_ptr = CreateDMatrix(10, 10, 0);
|
|
std::vector<std::shared_ptr<xgboost::DMatrix>> mat = {*mat_ptr};
|
|
auto learner = std::unique_ptr<Learner>(Learner::Create(mat));
|
|
|
|
// Test if `tree_method` can be set
|
|
learner->Configure({arg("tree_method", "approx")});
|
|
ASSERT_EQ(LearnerTestHookAdapter::GetUpdaterSequence(learner.get()),
|
|
"grow_histmaker,prune");
|
|
learner->Configure({arg("tree_method", "exact")});
|
|
ASSERT_EQ(LearnerTestHookAdapter::GetUpdaterSequence(learner.get()),
|
|
"grow_colmaker,prune");
|
|
learner->Configure({arg("tree_method", "hist")});
|
|
ASSERT_EQ(LearnerTestHookAdapter::GetUpdaterSequence(learner.get()),
|
|
"grow_fast_histmaker");
|
|
#ifdef XGBOOST_USE_CUDA
|
|
learner->Configure({arg("tree_method", "gpu_exact")});
|
|
ASSERT_EQ(LearnerTestHookAdapter::GetUpdaterSequence(learner.get()),
|
|
"grow_gpu,prune");
|
|
learner->Configure({arg("tree_method", "gpu_hist")});
|
|
ASSERT_EQ(LearnerTestHookAdapter::GetUpdaterSequence(learner.get()),
|
|
"grow_gpu_hist");
|
|
#endif
|
|
|
|
delete mat_ptr;
|
|
}
|
|
|
|
} // namespace xgboost
|