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

[Feature] Adding support for probability output mode in ml.rf_to_strings #550

Merged
merged 2 commits into from
Jul 1, 2021

Conversation

KMarkert
Copy link
Member

@KMarkert KMarkert commented Jul 1, 2021

This pull request adds a feature to ml.rf_to_strings which allows users to create tree strings where the output is probability of class. It should be noted that only binary classification problems can be supported at this time.

Changes:

  • ml.py: added output_mode option for "PROBABILITY" which formats the tree strings with probability values

Example usage:

Closes #522

@giswqs
Copy link
Member

giswqs commented Jul 1, 2021

Awesome addiction! Thank you.

@giswqs giswqs merged commit ff20723 into master Jul 1, 2021
@lgtm-com
Copy link

lgtm-com bot commented Jul 1, 2021

This pull request introduces 1 alert when merging d509136 into 02dc20d - view on LGTM.com

new alerts:

  • 1 for Unreachable code

@giswqs giswqs deleted the feature/rf-proba-outputs branch December 25, 2021 20:46
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

Successfully merging this pull request may close these issues.

Allow probability outputmode on RF classifier
2 participants