Skip to content
Snippets Groups Projects
  • Xiangrui Meng's avatar
    50a4fa77
    [SPARK-3443][MLLIB] update default values of tree: · 50a4fa77
    Xiangrui Meng authored
    Adjust the default values of decision tree, based on the memory requirement discussed in https://github.com/apache/spark/pull/2125 :
    
    1. maxMemoryInMB: 128 -> 256
    2. maxBins: 100 -> 32
    3. maxDepth: 4 -> 5 (in some example code)
    
    jkbradley
    
    Author: Xiangrui Meng <meng@databricks.com>
    
    Closes #2322 from mengxr/tree-defaults and squashes the following commits:
    
    cda453a [Xiangrui Meng] fix tests
    5900445 [Xiangrui Meng] update comments
    8c81831 [Xiangrui Meng] update default values of tree:
    50a4fa77
    History
    [SPARK-3443][MLLIB] update default values of tree:
    Xiangrui Meng authored
    Adjust the default values of decision tree, based on the memory requirement discussed in https://github.com/apache/spark/pull/2125 :
    
    1. maxMemoryInMB: 128 -> 256
    2. maxBins: 100 -> 32
    3. maxDepth: 4 -> 5 (in some example code)
    
    jkbradley
    
    Author: Xiangrui Meng <meng@databricks.com>
    
    Closes #2322 from mengxr/tree-defaults and squashes the following commits:
    
    cda453a [Xiangrui Meng] fix tests
    5900445 [Xiangrui Meng] update comments
    8c81831 [Xiangrui Meng] update default values of tree: