Feature weights (#5962)

2020-08-18 19:55:41 +08:00
parent a418278064
commit 4d99c58a5f
25 changed files with 509 additions and 104 deletions
--- a/tests/cpp/tree/test_gpu_hist.cu
+++ b/tests/cpp/tree/test_gpu_hist.cu
@@ -204,12 +204,11 @@ TEST(GpuHist, EvaluateRootSplit) {
  ASSERT_EQ(maker.hist.Data().size(), hist.size());
  thrust::copy(hist.begin(), hist.end(),
    maker.hist.Data().begin());
+  std::vector<float> feature_weights;

-  maker.column_sampler.Init(kNCols,
-    param.colsample_bynode,
-    param.colsample_bylevel,
-    param.colsample_bytree,
-    false);
+  maker.column_sampler.Init(kNCols, feature_weights, param.colsample_bynode,
+                            param.colsample_bylevel, param.colsample_bytree,
+                            false);

  RegTree tree;
  MetaInfo info;