Skip to content

[FEA] Add support for Feature Importance Scores for cuML Random Forest #7396

@csadorf

Description

@csadorf

Description

Add feature importance scores to cuML's Random Forest implementation, similar to sklearn's feature_importances_ attribute. For reference, see the sklearn example here: https://scikit-learn.org/stable/auto_examples/ensemble/plot_forest_importances.html.

Rationale

Feature importance scores would enable rapid assessment of feature importance as a precursor to exploring other candidate models, leveraging cuML's GPU-accelerated training speed.

Original issue: #3361

Metadata

Metadata

Assignees

Labels

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions