feat: improvements for federated learning and QoL

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: improvements for federated learning and QoL #137

Open matteo4diani opened 9 months ago

matteo4diani commented 9 months ago

Logger
Pre-Commit job with auto-format
Remove dead code
Centralize duplicated code for losses and utils
(dcph) Dropout
(dcph) Activations
(dcph) Optimizers
(dcph) Biases
(dcph) Early init
(dcph) Tests

chiragnagpal commented 9 months ago

The diffs don't resolve with the previous commit. Something is broken.

Please only add diffs for federated learning, QoL improvements should be a separate commit.

Thanks.. !

matteo4diani commented 9 months ago

This PR is just a container to keep track of changes I'm making in my fork, I will propose them in separate PRs. If we can re-open and add a "Do not merge" label it's useful for me to test CI on this repo. PS: The diff is broken because I enforced formatting, something for which I'll open a new PR.