Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Expose feature importances of EvoTreeClassifier #745

Closed
casasgomezuribarri opened this issue Feb 28, 2021 · 2 comments
Closed

Expose feature importances of EvoTreeClassifier #745

casasgomezuribarri opened this issue Feb 28, 2021 · 2 comments

Comments

@casasgomezuribarri
Copy link

casasgomezuribarri commented Feb 28, 2021

Is your feature request related to a problem? Please describe.
feature importance is not reported for EvoTreeClassifier models (pkg = EvoTrees.jl) (see this issue)

Describe the solution you'd like
Similar behaviour than with other models, feature importance returned when report(mach) is called.

Additional context
There is a function in EvoTrees.jl (https://github.com/Evovest/EvoTrees.jl/blob/master/src/importance.jl) that does this already.

@ablaom

@jeremiedb
Copy link

Thanks for raising this. I'll add the features importance reporting to the EvoTrees MLJ wrapper in the upcoming days.

@ablaom
Copy link
Member

ablaom commented Mar 2, 2021

@jeremiedb The format for the feature importances, within the report, is up to you, but you may want to keep this in mind: #747

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants