search

jaredleekatzman / DeepSurv

DeepSurv is a deep learning approach to survival analysis.
MIT License
584 stars 173 forks source link

question about training #24

Closed zakizhou closed 6 years ago

zakizhou commented 6 years ago

Thanks for the great work @jaredleekatzman, but I have a question about training of DeepSurv. Is there any "batch" split from train set when training or just a whole epoch of example are feed to the network directly?

jaredleekatzman commented 6 years ago

@zakizhou Mini-batch isn't implemented, so DeepSurv does a whole batch per epoch