没有softmax的图像分类器

时间:2018-03-07 17:30:49

标签: softmax turi

我使用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图层?

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

1 个答案:

答案 0 :(得分:0)

以下是访问模型属性的方法。它意味着特征,系数等。

Turi Documentation

如果有帮助请告诉我