我正在使用 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 层?
过去我曾提供过类似的模型,但现在我需要生成自己的模型。我已经阅读了文档并查看了源代码(如果需要我愿意修改),但我对机器学习的了解还不够,无法弄清楚。