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

Minor refactor #66

Closed GabrielGimenez closed 3 months ago

GabrielGimenez commented 1 year ago

Status

READY

Todo list

Background context

Consolidating XGBoost step of the model in a FeatureExtractor class, to simplify the development of new features later Added categorical feature support

Remove XGBoost default parameters, as it may surprise users as we were using different defaults over xgboost

Starting adding types to functions