search

square / pysurvival

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

Feature importance in Forest Models #37

Open elopezfune opened 3 years ago

elopezfune commented 3 years ago

To train any forest model with categorical variables it is needed first to convert them to dummy. After training, to get the feature importance, how does one get only one importance for only one variable, instead of for each class-factor?