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

Get direction of feature weights for binary classification #730

Open
fiaosborne opened this issue Jul 3, 2024 · 2 comments
Open

Get direction of feature weights for binary classification #730

fiaosborne opened this issue Jul 3, 2024 · 2 comments

Comments

@fiaosborne
Copy link

Hello, I am using autoML to perform a binary classification task. Would someone be able to advise me on how I could create a visual similar to the mean feature importance chart but that also displays weather the feature indicates a weighting towards the positive or negative class. This would be really helpful for my interpretation of my model.

There are packages within scikit learn that do this but I do not know how to extract a particular model from automl to then use in scikit learn.

@pplonski
Copy link
Contributor

pplonski commented Jul 8, 2024

Hi @fiaosborne,

Thanks for question. What method would you like to use for interpretation? I think the easiest way would be to use AutoML to search for optimal parameters and then train single model with scikit-learn.

@fiaosborne
Copy link
Author

I am attempting to generate partial dependence plots to understand the directionality of each features influence on my model

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