[ https://issues.apache.org/jira/browse/SPARK-17281?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=15492931#comment-15492931 ]
Weichen Xu commented on SPARK-17281: ------------------------------------ because currently, the AFTSuvivalRegression use `treeAggregate` and the `depth` default is 2. but, in many case the AFTSuvivalRegression only trains on low dimemsion data, in such case, the `treeAggregateDepth` set to 1 has better performance. so, I think add this parameter is useful. thanks! > Add treeAggregateDepth parameter for AFTSurvivalRegression > ---------------------------------------------------------- > > Key: SPARK-17281 > URL: https://issues.apache.org/jira/browse/SPARK-17281 > Project: Spark > Issue Type: Improvement > Components: ML, MLlib > Reporter: Weichen Xu > Priority: Minor > Original Estimate: 24h > Remaining Estimate: 24h > > Add treeAggregateDepth parameter for AFTSurvivalRegression. -- 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