Базовая научная модель coreML, возвращающая целые вероятности - PullRequest
0 голосов
/ 09 октября 2018

Я обучил модель scikit learn RandomForestClassifier и преобразовал ее в формат iOS coreML с coremltools.converters.sklearn.Модель имеет 4 класса в целевой переменной.При обучении целевая переменная равна string, а не int.

. Когда я выполняю модель на iOS, выходные данные вероятностей класса составляют 4 целых значения.Как мне это интерпретировать?Это вероятность?Я ожидал бы вероятности, которые составляют 1.

Примечание: Когда я делаю то же самое с моделью 2 классов (целевая переменная int), я получаю вероятности класса, которые составляют 1.

Есть идеи, как решить эту проблему?

...