search

square / pysurvival

Open source package for Survival Analysis modeling
https://www.pysurvival.io/
Apache License 2.0
350 stars 106 forks source link

Is there a way to manually set the the time buckets? #13

Open AlexFuster opened 4 years ago

AlexFuster commented 4 years ago

Hey

I would like to fix the time buckets before training, but I don't think the library supports that for now.
I tried setting the attributes times and time_buckets, but that doesn't work, as the model calculates its own ones during training.
I think that it would be useful to be able to specify that