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

Add the ability to calculate importance by class for multiclass classification #154

Open
see24 opened this issue Aug 22, 2018 · 2 comments

Comments

@see24
Copy link

see24 commented Aug 22, 2018

This is not possible in the old gbm but I recently wrote a function to calculate it. It can yield very different results for the most important variables for different classes. See this question and my answer for the code
Hopefully you would just need to translate this from my dplyr/purrr version to base to implement it. If it doesn't make sense to do this this way please let me know!

@ayushnoori
Copy link

Has variable importance for multiclass classification officially been implemented yet? If not, @see24 did you find any issues with the code which you posted on SO?

@see24
Copy link
Author

see24 commented Feb 22, 2021

I did continue using this code and I published an article that used the by class variable importance. In this article the importance of different variables for different classes mostly aligned with our ecological expectations so I felt pretty good about it. However, nobody with in depth knowledge of the gbm package or the statistics behind it has reviewed the code as far as I know.

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

2 participants