How do I get the corresponding feature importance of every variable in a GBT Classifier model in pyspark
Asked
Active
Viewed 4,024 times
1 Answers
0
From spark 2.0+ (here) You have the attribute:
model.featureImportances
This will give a sparse vector of feature importance for each column/ attribute
Community
- 1
- 1
pratiklodha
- 1,013
- 12
- 20