Skip to content

ENH: Choose feature/features for use in plotting feature importance #638

@andersbogsnes

Description

@andersbogsnes

Is your feature request related to a problem? Please describe.
Sometimes it can take a while to calculate feature importance for the entire dataframe - can we select only some features to permute instead of permuting every feature?

Describe the solution you'd like
Should be able to pass a feature name/list of feature names to result.plot.feature_importance and see the permutation results for those

Metadata

Metadata

Assignees

No one assigned

    Labels

    enhancementNew feature or request

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions