[jira] [Resolved] (SPARK-3043) DecisionTree aggregation is inefficient

["Xiangrui Meng (JIRA)" <ji...@apache.org>]

     [ https://issues.apache.org/jira/browse/SPARK-3043?page=com.atlassian.jira.plugin.system.issuetabpanels:all-tabpanel ]

Xiangrui Meng resolved SPARK-3043.
----------------------------------
          Resolution: Fixed
       Fix Version/s: 1.2.0
    Target Version/s: 1.2.0  (was: 1.1.0)

> DecisionTree aggregation is inefficient
> ---------------------------------------
>
>                 Key: SPARK-3043
>                 URL: https://issues.apache.org/jira/browse/SPARK-3043
>             Project: Spark
>          Issue Type: Improvement
>          Components: MLlib
>    Affects Versions: 1.1.0
>            Reporter: Joseph K. Bradley
>            Assignee: Joseph K. Bradley
>             Fix For: 1.2.0
>
>
> 2 major efficiency issues in computation and storage:
> (1) DecisionTree aggregation involves reshaping data unnecessarily.
> E.g., the internal methods extractNodeInfo() and getBinDataForNode() involve reshaping the data multiple times without real computation.
> (2) DecisionTree splits and aggregate bins can include many unused bins/splits.
> The same number of splits/bins are used for all features.  E.g., if there is a continuous feature which uses 100 bins, then there will also be 100 bins allocated for all binary features, even though only 2 are necessary.



--
This message was sent by Atlassian JIRA
(v6.3.4#6332)

---------------------------------------------------------------------
To unsubscribe, e-mail: issues-unsubscribe@spark.apache.org
For additional commands, e-mail: issues-help@spark.apache.org