why `batch` is not appropriate in survival analysis? about tfdeepsurv

[liupei101]

To put it straightforwardly, if we dive into the loss function (partial likelihood function), survival analysis is a variant of ranking task.

Since we did not find any reference paper to support running batch training in survival analysis, the code remained as you saw.

Thx for your feedbacks! TFDeepSurv has implemented the pipeline in survival analysis, but we can easily find the shorts. We need to add functionality that supports:

• GPU training
• mini-batch training (experimentally)
  to speed up the training phase!

In the near future, above plans will not be implemented. Sorry about it!

PRs are welcomed!

[bnuzyc91]

Thank you so much for the explanation. Now I agree that mini-batch may not be a good idea for the survival analysis especially in the case of a rare event (the chance of developing the outcome is low).