Skip to content

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

3 participants