GBT Classifier Pyspark中每个变量的特征重要性

时间:2018-02-09 11:55:20

标签: apache-spark pyspark

如何在pyspark

中获取GBT分类器模型中每个变量的相应特征重要性

1 个答案:

答案 0 :(得分:0)

来自spark 2.0+(here)你有以下属性:

model.featureImportances

这将为每个列/属性

提供特征重要性的稀疏向量