Skip to content
Closed
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Finished a TODO for comments in Strategy.scala
  • Loading branch information
CBribiescas committed Jun 20, 2021
commit 4bb58f6e0d0932bcaf95e127d5a44ae52271e5f0
Original file line number Diff line number Diff line change
Expand Up @@ -55,7 +55,8 @@ import org.apache.spark.mllib.tree.impurity.{Entropy, Gini, Impurity, Variance}
* @param minInfoGain Minimum information gain a split must get. Default value is 0.0.
* If a split has less information gain than minInfoGain,
* this split will not be considered as a valid split.
* @param pruneTree <TODO>
* @param pruneTree If this is true, the final training tree will undergo a pruning in which
* nodes with the same classifications are merged.
* @param maxMemoryInMB Maximum memory in MB allocated to histogram aggregation. Default value is
* 256 MB. If too small, then 1 node will be split per iteration, and
* its aggregates may exceed this size.
Expand Down