Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[ENH] adapter to scikit-survival, all distributional survival regressors interfaced #237

Merged
merged 32 commits into from
Apr 8, 2024

Conversation

fkiraly
Copy link
Collaborator

@fkiraly fkiraly commented Apr 6, 2024

Adds an adapter to scikit-survival, exposing array-like survival functions as Empirical distributions in predict_proba.

Adds all models from scikit-survival which are capable of full distributional predictions:

  • CoxPhSurvivalAnalysis
  • CoxNetSurvivalAnalysis
  • SurvivalTree
  • GradientBoostingSurvivalAnalysis
  • ComponentwiseGradientBoostingSurvivalAnalysis
  • RandomSurvivalForest
  • ExtraSurvivalTrees

The remaining models, e.g., SVM, are not capable of distributional predictions.

@fkiraly fkiraly added enhancement interfacing algorithms Interfacing existing algorithms/estimators from third party packages module:survival&time-to-event module for time-to-event prediction aka survival prediction labels Apr 6, 2024
@fkiraly fkiraly marked this pull request as ready for review April 6, 2024 23:02
@fkiraly fkiraly changed the title [ENH] adapter to scikit-survival [ENH] adapter to scikit-survival, all distributional survival regressors interfaced Apr 6, 2024
@fkiraly fkiraly merged commit 86ec7d4 into main Apr 8, 2024
34 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement interfacing algorithms Interfacing existing algorithms/estimators from third party packages module:survival&time-to-event module for time-to-event prediction aka survival prediction
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant