Skip to content

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?

3 participants