trees

Type Members

1. sealed abstract class FeatureSubsetStrategy extends AnyRef

   

   The number of features to consider for splits at each tree node.

   The number of features to consider for splits at each tree node. Supported options:

   • Auto: Choose automatically for task: If numTrees == 1, set to All If numTrees > 1 (forest), set to Sqrt for classification and to OneThird for regression.
   • All: use all features
   • OneThird: use 1/3 of the features
   • Sqrt: use sqrt(number of features)
   • Log2: use log2(number of features)
   • Ratio: use (ratio * number of features) features
   • NumberOfFeatures: use numberOfFeatures features. (default = Auto)

   These various settings are based on the following references:

   • log2: tested in Breiman (2001)
   • sqrt: recommended by Breiman manual for random forests
   • The defaults of sqrt (classification) and onethird (regression) match the R randomForest package.

   See also

   Breiman manual for random forests

   Breiman (2001)