[
https://issues.apache.org/jira/browse/SPARK-8518?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=14700302#comment-14700302
]
Meihua Wu commented on SPARK-8518:
----------------------------------
[~yanbo] Thank you very much for the update!
The loss function and gradient are different for events and censor. So we will
need to have a column in the data frame to indicate whether an individual
record is an event or censored. I suppose we will need to define a Param for
"eventCol" using code gen and mix it into the AFTRegressionParams.
cc [~mengxr]
> Log-linear models for survival analysis
> ---------------------------------------
>
> Key: SPARK-8518
> URL: https://issues.apache.org/jira/browse/SPARK-8518
> Project: Spark
> Issue Type: New Feature
> Components: ML
> Reporter: Xiangrui Meng
> Assignee: Yanbo Liang
> Original Estimate: 168h
> Remaining Estimate: 168h
>
> We want to add basic log-linear models for survival analysis. The
> implementation should match the result from R's survival package
> (http://cran.r-project.org/web/packages/survival/index.html).
> Design doc from [~yanboliang]:
> https://docs.google.com/document/d/1fLtB0sqg2HlfqdrJlNHPhpfXO0Zb2_avZrxiVoPEs0E/pub
--
This message was sent by Atlassian JIRA
(v6.3.4#6332)
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
