No observation in time bucket

loft-br / xgboost-survival-embeddings

Improving XGBoost survival analysis with embeddings and debiased estimators

https://loft-br.github.io/xgboost-survival-embeddings/

Apache License 2.0

321 stars 53 forks source link

No observation in time bucket #37

Open GabrielGimenez opened 3 years ago

GabrielGimenez commented 3 years ago

When there's no observation in a time bucket in BCE we get a value error ValueError: Error: No observations in a time bucket It would be nice to see in which time bucket was the error

igorkf commented 3 years ago

I got this warning when fitting a XGBSEDebiasedBCE model: Warning: Only one class found in a time bucket
Can you explain what it means, please?

GabrielGimenez commented 3 years ago

@igorkf This warning occurs when a time bucket you don't have any event or only events, so that specific logistic regression for that time bucket will only predict 0 or 1

more details on XGBSEDebiasedBCE works https://loft-br.github.io/xgboost-survival-embeddings/how_xgbse_works.html