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.

> 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

To unsubscribe, e-mail: issues-unsubscr...@spark.apache.org
For additional commands, e-mail: issues-h...@spark.apache.org

Reply via email to