autonlab / auton-survival

Auton Survival - an open source package for Regression, Counterfactual Estimation, Evaluation and Phenotyping with Censored Time-to-Events
http://autonlab.github.io/auton-survival
MIT License
316 stars 74 forks source link

Memory efficient patience #43

Open Jeanselme opened 3 years ago

Jeanselme commented 3 years ago

This avoids to keep all states in memory. Apply same strategy for pre training