Add categorical data support to GPU Hist. (#6164)

2020-09-29 11:27:25 +08:00
parent 798af22ff4
commit 444131a2e6
9 changed files with 306 additions and 103 deletions
--- a/src/tree/gpu_hist/evaluate_splits.cuh
+++ b/src/tree/gpu_hist/evaluate_splits.cuh
@@ -18,6 +18,7 @@ struct EvaluateSplitInputs {
  GradientSumT parent_sum;
  GPUTrainingParam param;
  common::Span<const bst_feature_t> feature_set;
+  common::Span<FeatureType const> feature_types;
  common::Span<const uint32_t> feature_segments;
  common::Span<const float> feature_values;
  common::Span<const float> min_fvalue;