I have trained a scikit learn RandomForestClassifier model and converted it to the iOS coreML format with coremltools.converters.sklearn. The model has 4 classes in the target variable. When trained the target variable is string as opposed to int.
When I execute the model on iOS the class probabilities output are 4 integer values. How do I interpret these? Are they likelihoods? I would have expected probabilities that sum to 1.
Note: When I do the same with a 2 class model (target variable is int) I get the class probabilities that sum to 1.
Any ideas on how to resolve this?
from coreML scikit learn model returning integer probabilities
No comments:
Post a Comment