[
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: [email protected]
For additional commands, e-mail: [email protected]
