Losses

torch_survival.losses.cox_neg_log_likelihood(risk: Tensor, event: Tensor, time: Tensor, sort: bool = True) → Tensor

Cox negative log partial likelihood. From DeepSurv: https://doi.org/10.1186/s12874-018-0482-1

Parameters:

• risk (torch.Tensor, of shape (..., n_samples)) – Risks estimated by model

• event (torch.Tensor, of shape (..., n_samples)) – Event indicator denoting whether time is of observed event or dropout

• time (torch.Tensor, of shape (..., n_samples)) – Time of either observed event or dropout

• sort (bool) – Whether to sort by time, otherwise assumes pre-sorted ground truth times and events