Fix GPU RF (#6755)

* Fix sampling.
2021-03-17 06:23:35 +08:00
parent 1a73a28511
commit 4f75f514ce
4 changed files with 26 additions and 9 deletions
--- a/tests/python-gpu/test_gpu_with_sklearn.py
+++ b/tests/python-gpu/test_gpu_with_sklearn.py
@@ -33,4 +33,8 @@ def test_gpu_binary_classification():


 def test_boost_from_prediction_gpu_hist():
-    cpu_test = twskl.run_boost_from_prediction('gpu_hist')
+    twskl.run_boost_from_prediction('gpu_hist')
+
+
+def test_num_parallel_tree():
+    twskl.run_boston_housing_rf_regression("gpu_hist")