-
Notifications
You must be signed in to change notification settings - Fork 23
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
Add Dynamic DeepHit (time-to-event) #39
Conversation
There are still some additional metrics that need to be added as per #9 - will add that in future PR |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Great work! LGTM!
There is a discussion about time-to-event naming: There is dedicated literature for time-to-event regression, which approximates the date of the event. And is different from the current models you have.
Otherwise, survival models approximate the probability of risk(class 1) or survival(class 0) for multiple horizons. In AutoPrognosis I am approximating the probability of risk(same as here), so these models are named risk_estimation
models.
So, it should be clear that you don't approximate the date of the event, neither the survival probability. But the risk probability.
Description
Closes #22
DynamicDeepHitTimeToEventAnalysis
and the infrastructure and models needed for that.Separately:
DataLoader
class.Affected Dependencies
scikit-learn
added.How has this been tested?
Checklist