iOS上的ML模型意外行为(Swift)

时间:2017-11-13 19:48:01

标签: python ios swift coreml

当我将ML模型从Python集成到Swift时,我遇到了一个奇怪的问题。

型号: 卷积NN

输入:1000浮动

输出(在Python中):一对2浮动

步骤:

1)准备好集成模型(从Python创建* .mlmodel文件)

2)使用官方文档将其添加到我的应用程序中

3)尝试使用生成的代码

效果很好,但结果出乎意料:

输入:1000个浮点数 - 与Python代码相同

输出:1000对两个花车

如果我可以预期输出与Python相同的输出? 如果它可以帮助 - 我将提供必要的代码

更新1

模型导出代码(Python):

import coremltools
coreml_model = coremltools.converters.keras.convert(autoencoder, input_names="input_1", output_names="dense_3/Softmax")
coreml_model.save('validation_model.mlmodel')

模型在Xcode中的外观截图 - https://prnt.sc/ha14ih

0 个答案:

没有答案