Spark ML - Feature Importance for Tree Models

Usage

ml_tree_feature_importance(sc, model)

Arguments

sc
An active spark context
model
An ml_model object of class decision trees (>1.5.0), random forest (>2.0.0), or GBT (>2.0.0)

Value

A sorted data frame with feature labels and their relative importance.

Description

Spark ML - Feature Importance for Tree Models