-
Notifications
You must be signed in to change notification settings - Fork 3
Adding feature_importances to report for DecisionTree models #28
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
Conversation
I can see here that you are using |
|
Are you suggesting I add a new hyperparamter to the structs in |
Yes. |
|
Okay, now we've got the ability to choose from |
Don't worry about permutation importance for now. That would added later to MLJBase. |
|
Okay, sounds good. Is there anything else that needs to be done? |
Codecov Report
@@ Coverage Diff @@
## dev #28 +/- ##
==========================================
+ Coverage 86.20% 86.88% +0.67%
==========================================
Files 1 1
Lines 87 122 +35
==========================================
+ Hits 75 106 +31
- Misses 12 16 +4
Help us with your feedback. Take ten seconds to tell us how you rate us. Have a feature suggestion? Share it here. |
OkonSamuel
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
@john-waczak. I have reviewed your PR and suggested changes.
…ue, and define MMI.feature_importances function
|
@OkonSamuel Okay, I think I have it updated to follow the format now. Apologies for the confusion; I was using EvoTrees.jl as an example and thought they needed to be in the report. I ended up adding features to the report of each model so they'd be available for the feature_importances function. I guess EvoTrees.jl will need to be updated next, right? |
Yes. We plan to make changes to several packages including EvoTrees.jl |
OkonSamuel
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks @john-waczak for all your effort. Just a few things that needs to be addressed and this PR would be ready for merge.
|
@OkonSamuel Glad to help! |
|
Okay, great to have this. I'll tag a new release. |
This update adds feature_importances to the report for each model from DecisionTrees.jl using the
impurity_importancefunction. I've followed the same output format as the feature_importances from EvoTrees.jl and added a simple test for each method to runtests.jl.