Uploaded image for project: 'Spark'
  1. Spark
  2. SPARK-29232

RandomForestRegressionModel does not update the parameter maps of the DecisionTreeRegressionModels underneath

    XMLWordPrintableJSON

    Details

    • Type: Bug
    • Status: Resolved
    • Priority: Major
    • Resolution: Fixed
    • Affects Version/s: 2.4.0
    • Fix Version/s: 3.0.0
    • Component/s: ML
    • Labels:
      None

      Description

      We trained a RandomForestRegressionModel, and tried to access the trees. Even though the DecisionTreeRegressionModel is correctly built with the proper parameters from random forest, the parameter map is not updated, and still contains only the default value. 

      For example, if a RandomForestRegressor was trained with maxDepth of 12, then accessing the tree information, extractParamMap still returns the default values, with maxDepth=5. Calling the depth itself of DecisionTreeRegressionModel returns the correct value of 12 though.

      This creates issues when we want to access each individual tree and build the trees back up for the random forest estimator.

        Attachments

          Issue Links

            Activity

              People

              • Assignee:
                huaxingao Huaxin Gao
                Reporter:
                jiaqig Jiaqi Guo
              • Votes:
                0 Vote for this issue
                Watchers:
                4 Start watching this issue

                Dates

                • Created:
                  Updated:
                  Resolved: