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

RandomForestRegressionModel does not update the parameter maps of the DecisionTreeRegressionModels underneath

    XMLWordPrintableJSON

Details

    • Bug
    • Status: Resolved
    • Major
    • Resolution: Fixed
    • 2.4.0
    • 3.0.0
    • ML
    • 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

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

              Dates

                Created:
                Updated:
                Resolved: