-
Notifications
You must be signed in to change notification settings - Fork 64
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
Classifier: output proba instead of category #155
Comments
For the case of a single category (defaulted or not, bought or not), yes! For the case of multiple categories (shopper type A,B,C, or D), you should be able to, but it will require a little bit of work. To do this, go to pySetup/makePredictions.py. So in the else case (problemType != 'category'), simply change classifier.predict to classifier.predict_proba. Now our ensemble algorithms will make probability predictions. Again, this means that when ensembler goes to assemble together the predictions from all our ensemble algorithms, it will fail. But that's ok. Simply find which of those algorithms was most accurate, and use the results found in that file. Again, this whole process is a bit of a hack for now. At some point I'll try to build this in as standard functionality in ensembler, but it will be increasing complexity substantially, and there are easier wins to focus on right now. If you're interested in taking a crack at building it in, I'd love that PR! Thanks for using this, and filing issues. Let me know if you have any other questions! |
Is it possible to get the proba of each category, and not the most probable one, when using classifiers?
The text was updated successfully, but these errors were encountered: