[ https://issues.apache.org/jira/browse/SPARK-3159?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=17293294#comment-17293294 ]
Carlos Bribiescas edited comment on SPARK-3159 at 3/2/21, 2:20 AM: ------------------------------------------------------------------- Would setting `setCanMergeChildren` in construction disable this? Or am i reading the PR wrong? was (Author: cbribiescas): Would setting `canMergeChildren` in construction disable this? Or am i reading the PR wrong? > Check for reducible DecisionTree > -------------------------------- > > Key: SPARK-3159 > URL: https://issues.apache.org/jira/browse/SPARK-3159 > Project: Spark > Issue Type: Improvement > Components: MLlib > Reporter: Joseph K. Bradley > Assignee: Alessandro Solimando > Priority: Minor > Fix For: 2.4.0 > > Attachments: image-2020-05-24-23-00-38-419.png > > > Improvement: test-time computation > Currently, pairs of leaf nodes with the same parent can both output the same > prediction. This happens since the splitting criterion (e.g., Gini) is not > the same as prediction accuracy/MSE; the splitting criterion can sometimes be > improved even when both children would still output the same prediction > (e.g., based on the majority label for classification). > We could check the tree and reduce it if possible after training. > Note: This happens with scikit-learn as well. -- This message was sent by Atlassian Jira (v8.3.4#803005) --------------------------------------------------------------------- To unsubscribe, e-mail: issues-unsubscr...@spark.apache.org For additional commands, e-mail: issues-h...@spark.apache.org