ml_feature_importances: Spark ML - Feature Importance for Tree Models

View source: R/ml_helpers.R

ml_feature_importancesR Documentation

Spark ML - Feature Importance for Tree Models

Description

Spark ML - Feature Importance for Tree Models

Usage

ml_feature_importances(model, ...)

ml_tree_feature_importance(model, ...)

Arguments

model

A decision tree-based model.

...

Optional arguments; currently unused.

Value

For ml_model, a sorted data frame with feature labels and their relative importance. For ml_prediction_model, a vector of relative importances.


rstudio/sparklyr documentation built on Sept. 18, 2024, 6:10 a.m.