Monday, 22 October 2018

coreML scikit learn model returning integer probabilities

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