当我将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