How can we get the importance and output trees?

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

How can we get the importance and output trees? #52

Closed sqf10000 closed 2 years ago

sqf10000 commented 2 years ago

XGBSEDebiasedBCE.plot_importance(xgbse_model) AttributeError: type object 'XGBSEDebiasedBCE' has no attribute 'plot_importance'

Can you add the plot_importance function in your xgbse?

GabrielGimenez commented 2 years ago

Hi! We are discussing how to make the internal xgboost model easier to use, but for now, you can get feature importance using xgbse_model.bst.get_score(importance_type='gain')