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
315 stars 74 forks source link

feat(dcph): enable federated learning with DeepCoxPH #134

Closed matteo4diani closed 9 months ago

matteo4diani commented 10 months ago

Solves #120

Federated learning is of great interest for survival analysis: to use the DeepCoxPH model in a FedML experiment we need a couple of features that this PR adds:

A couple of unrelated QoL improvements have been added to this PR to support its implementations, namely:

matteo4diani commented 10 months ago

@chiragnagpal please review 🙏🏼

matteo4diani commented 9 months ago

Integrated in https://github.com/autonlab/auton-survival/pull/137