Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Implement hist evaluator for multi-target tree. (#8908)

Browse Source
This commit is contained in:
Jiaming Yuan
2023-03-15 01:42:51 +08:00
committed by GitHub
parent 95e2baf7c2
commit 8685556af2
7 changed files with 416 additions and 76 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/cpp/tree/gpu_hist/test_evaluate_splits.cu
View File

@@ -304,7 +304,7 @@ void TestEvaluateSingleSplit(bool is_categorical) {
thrust::device_vector<bst_feature_t> feature_set = std::vector<bst_feature_t>{0, 1};
// Setup gradients so that second feature gets higher gain
auto feature_histogram = ConvertToInteger({ {-0.5, 0.5}, {0.5, 0.5}, {-1.0, 0.5}, {1.0, 0.5}});
auto feature_histogram = ConvertToInteger({{-0.5, 0.5}, {0.5, 0.5}, {-1.0, 0.5}, {1.0, 0.5}});
dh::device_vector<FeatureType> feature_types(feature_set.size(),
FeatureType::kCategorical);