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

[Bugfix] Fix for incorrect output labels from ml.rf_to_strings workflow #545

Merged
merged 2 commits into from
Jun 30, 2021

Conversation

KMarkert
Copy link
Member

This PR contains as a bug fix to correctly label the classification results when serializing a random forest model to strings for later use on EE. This also contains some support to infer output mode of the user provided classifier as well as allow users to explicitly define the output mode (used when assigning the correct output labels).

Changes:

  1. ml.py: updated value labels to be gathered from sklearn classifier and added checks for output mode

Example usage:

Closes #528

@giswqs
Copy link
Member

giswqs commented Jun 30, 2021

Looks good. Thanks for fixing the bug.

@giswqs giswqs merged commit 02dc20d into master Jun 30, 2021
@giswqs giswqs deleted the bugfix/rf_to_string-labels 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.

Strange classification when using rf_to_strings trees in EE RF?
2 participants