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

Add dropout #86

Open juancq opened 2 years ago

juancq commented 2 years ago

Adds a dropout parameter for adding dropout layers to the representation layers.