-4

How do I get the corresponding feature importance of every variable in a GBT Classifier model in pyspark

Alper t. Turker
  • 32,514
  • 8
  • 78
  • 112

1 Answers1

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