Skip to content

Commit cadd569

Browse files
author
qiping.lqp
committed
add api docs
1 parent 46b891f commit cadd569

File tree

1 file changed

+6
-0
lines changed
  • mllib/src/main/scala/org/apache/spark/mllib/tree/configuration

1 file changed

+6
-0
lines changed

mllib/src/main/scala/org/apache/spark/mllib/tree/configuration/Strategy.scala

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,12 @@ import org.apache.spark.mllib.tree.configuration.QuantileStrategy._
4949
* k) implies the feature n is categorical with k categories 0,
5050
* 1, 2, ... , k-1. It's important to note that features are
5151
* zero-indexed.
52+
* @param minInstancesPerNode Minimum number of nodes each child must have after split. Default value is 0.
53+
* If a split cause left or right child to have less than minInstancesPerNode,
54+
* this split will not be considered as a valid split.
55+
* @param minInfoGain Minimum information gain a split must get. Default value is 0.0.
56+
* If a split has less information gain than minInfoGain,
57+
* this split will not be considered as a valid split.
5258
* @param maxMemoryInMB Maximum memory in MB allocated to histogram aggregation. Default value is
5359
* 128 MB.
5460
*/

0 commit comments

Comments
 (0)