update forest params (#2286)

oneapi-src · Feb 28, 2023 · 32c31e5 · 32c31e5
1 parent b3b8ed5
commit 32c31e5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/examples/oneapi/dpc/source/decision_forest/df_reg_hist_batch.cpp b/examples/oneapi/dpc/source/decision_forest/df_reg_hist_batch.cpp
@@ -42,7 +42,7 @@ void run(sycl::queue& q) {
 
     const auto df_desc =
         df::descriptor<float, df::method::hist, df::task::regression>{}
-            .set_tree_count(100)
+            .set_tree_count(150)
             .set_features_per_node(0)
             .set_min_observations_in_leaf_node(1)
             .set_error_metric_mode(df::error_metric_mode::out_of_bag_error |

diff --git a/samples/oneapi/dpc/ccl/sources/decision_forest_cls_hist_distr_ccl.cpp b/samples/oneapi/dpc/ccl/sources/decision_forest_cls_hist_distr_ccl.cpp
@@ -57,7 +57,7 @@ void run(sycl::queue &queue) {
     const auto df_desc =
         df::descriptor<float, df::method::hist, df::task::classification>{}
             .set_class_count(5)
-            .set_tree_count(10)
+            .set_tree_count(50)
             .set_features_per_node(x_train.get_column_count())
             .set_min_observations_in_leaf_node(8)
             .set_min_observations_in_split_node(16)