Классификатор изображений без softmax

Я использую Turi Create для создания классификатора изображений Core ML следующим образом:

import turicreate as tc
data =  tc.SFrame('data.sframe')
model = tc.image_classifier.create(data, target='label')
model.export_coreml('classifier.mlmodel')

Однако модель дает мне значения достоверности softmax, и в моем приложении мне нужны значения pre-softmax (я рассчитаю softmax в приложении). Есть ли способ отключить слой softmax?

Раньше у меня были подобные модели, но теперь мне нужно создать свою собственную. Я прочитал документы и просмотрел исходный код (который я готов изменить, если это необходимо), но я недостаточно знаю о машинном обучении, чтобы понять это.


person Timothy Moose    schedule 07.03.2018    source источник


Ответы (1)