Core Ml Model, labels probabilities returning nan

Question

Hi, I have a core ml model that when I try to print the:

Code Block modelPrediction?.labelProbability

which is of type

Code Block String:Double

and contains all the features with their corresponding probabilities, the value of type double comes with nan

rest = nan
right = nan
up = nan

Sometimes restarting makes it work again.
Sometimes it can take a lot of restarts to start working again.
Even when deleting the app and installing again the same thing happens.
Also tried changing the deployment version but didn't seem to fix it.

Any help is appreciated.

2.2k

Posted by

jessylol

Reply

Add a Comment

Answer 1

Forgot to mention, the core ml model i'm using is a activity classifier

Posted by

jessylol

Add a Comment

Answer 2

I finally found what the problem was.
Just had to start my stateIn multi array with full of zeros and restart the state after not needing it.
After this no more label probability came with nan.

Posted by

jessylol

Add a Comment

Answer 3

May I know the exact solution of it. I am getting the similar issue.

Posted by

asantha

Add a Comment

Answer 4

I have the same problem. Exact solution would be very much appreciated.

Posted by

skybrick

Add a Comment

Answer 5

hey, i have the same issue, did you had any success in finding a solution?

Posted by

lexiestlexie

Add a Comment

Answer 6

I fixed this issue as well by following @jessylol suggestion. All you have to do is declare the stateIn param like this


 let array = Array(repeating: 0, count: 400)
 let stateIn = try MLMultiArray(array)

Posted by

Brunohenriquesk1

Add a Comment

Core Ml Model, labels probabilities returning nan

Replies