我使用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置信度值,在我的应用程序中,我需要之前的softmax值(我将在应用程序中计算softmax)。有没有办法关闭softmax层?
我以前也有这样的模型提供给我,但现在我需要生成我自己的模型。我已经阅读了文档和源代码(如果需要,我愿意修改),但我对机器学习的了解还不够多,无法弄清楚。
发布于 2018-03-08 19:24:37
https://stackoverflow.com/questions/49157961
复制相似问题