Skip to content

Commit 3717f6d

Browse files
committed
More detail on boosted tree
1 parent 212765f commit 3717f6d

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

man/rmd/boost-tree.Rmd

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,7 @@ get_defaults_boost_tree <- function() {
6969
"boost_tree", "spark", "tree_depth", "max_depth", get_arg("sparklyr", "ml_gradient_boosted_trees", "max_depth"),
7070
"boost_tree", "spark", "trees", "max_iter", get_arg("sparklyr", "ml_gradient_boosted_trees", "max_iter"),
7171
"boost_tree", "spark", "learn_rate", "step_size", get_arg("sparklyr", "ml_gradient_boosted_trees", "step_size"),
72-
"boost_tree", "spark", "mtry", "feature_subset_strategy", get_arg("sparklyr", "ml_gradient_boosted_trees", "feature_subset_strategy"),
72+
"boost_tree", "spark", "mtry", "feature_subset_strategy", "see below",
7373
"boost_tree", "spark", "min_n", "min_instances_per_node", get_arg("sparklyr", "ml_gradient_boosted_trees", "min_instances_per_node"),
7474
"boost_tree", "spark", "loss_reduction", "min_info_gain", get_arg("sparklyr", "ml_gradient_boosted_trees", "min_info_gain"),
7575
"boost_tree", "spark", "sample_size", "subsampling_rate", get_arg("sparklyr", "ml_gradient_boosted_trees", "subsampling_rate"),
@@ -79,3 +79,4 @@ get_defaults_boost_tree <- function() {
7979
parsnip::convert_args("boost_tree")
8080
```
8181

82+
For spark, the default `mtry` is the square root of the number of predictors for classification, and one-third of the predictors for regression.

0 commit comments

Comments
 (0)