[ https://issues.apache.org/jira/browse/SPARK-3724?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=15213517#comment-15213517 ]
Yong Tang commented on SPARK-3724: ---------------------------------- I can work on this one. Will create a PR soon. > RandomForest: More options for feature subset size > -------------------------------------------------- > > Key: SPARK-3724 > URL: https://issues.apache.org/jira/browse/SPARK-3724 > Project: Spark > Issue Type: Improvement > Components: MLlib > Reporter: Joseph K. Bradley > Priority: Minor > > RandomForest currently supports using a few values for the number of features > to sample per node: all, sqrt, log2, etc. It should support any given value > (to allow model search). > Proposal: If the parameter for specifying the number of features per node is > not recognized (as “all”, “sqrt”, etc.), then it will be parsed as a > numerical value. The value should be either (a) a real value in [0,1] > specifying the fraction of features in each subset or (b) an integer value > specifying the number of features in each subset. -- This message was sent by Atlassian JIRA (v6.3.4#6332) --------------------------------------------------------------------- To unsubscribe, e-mail: issues-unsubscr...@spark.apache.org For additional commands, e-mail: issues-h...@spark.apache.org