Trubles with model saving

Hi, I try to save my trained sklearn sgb model. I do the next:

import coremltools

coreml_model = coremltools.converters.sklearn.convert(xgb_model)

coreml_model.save('XGB.mlmodel')


But when I open it in my Xcode project, I see that input type is MLMultuArray<Invalid, 55000>, it should be an array of Double with length 55000. It seems, that because of that model doesn't work in the app. Could you please help me to tackle with it?

Replies

It also shows "Invalid" for my outputs, but my models work and the output is a MLMultuArray of doubles. Maybe it's just a display bug.