Spark ML - Feature Importance for Tree Models
R/ml_helpers.R
ml_feature_importances
Description
Spark ML - Feature Importance for Tree Models
Usage
ml_feature_importances(model, ...)
ml_tree_feature_importance(model, ...) Arguments
| Arguments | Description |
|---|---|
| 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.