sparklyr (version 0.3.11)

ml_tree_feature_importance: Spark ML - Feature Importance for Tree Models

Description

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.