Я обучил модель scikit learn RandomForestClassifier
и преобразовал ее в формат iOS coreML с coremltools.converters.sklearn
.Модель имеет 4 класса в целевой переменной.При обучении целевая переменная равна string
, а не int
.
. Когда я выполняю модель на iOS, выходные данные вероятностей класса составляют 4 целых значения.Как мне это интерпретировать?Это вероятность?Я ожидал бы вероятности, которые составляют 1.
Примечание: Когда я делаю то же самое с моделью 2 классов (целевая переменная int
), я получаю вероятности класса, которые составляют 1.
Есть идеи, как решить эту проблему?