xgboost-survival-embeddings icon indicating copy to clipboard operation
xgboost-survival-embeddings copied to clipboard

Published 20 hours ago • verified publisher icon loft-br

Minor refactor

Open GabrielGimenez opened this issue 1 year ago • 0 comments

Status

READY

Todo list

  • [X] Documentation
  • [X] Tests added and passed

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

GabrielGimenez avatar Mar 01 '23 16:03 GabrielGimenez