我使用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图层?
我过去曾经提供过这样的模型,但现在我需要创建自己的模型。我已经阅读了文档并查看了源代码(如果需要,我愿意修改),但我对机器学习知之甚少。