问如何使用coreML转换为模型输出指定标签名称
EN

Stack Overflow用户

提问于 2018-09-03 08:13:18

回答 1查看 383关注 0票数 0

我使用keras构建了一个模型，并希望使用以下函数将其转换为coreML：

import coremltools
coreml_model = coremltools.converters.keras.convert(model)
coreml_model.save(‘myModel’)

我的模型的输出是一个10个神经元层来预测10个类别。我的问题是，我想给出与每个神经元classA、classB等相关的标签名。

文档显示了许多参数(https://apple.github.io/coremltools/generated/coremltools.converters.keras.convert.html)，但我不知道使用哪个参数：output_names、predicted_feature_name或predicted_probabilities_output

回答已采纳

发布于 2018-09-03 08:16:45

算了吧..。我只是没有正确阅读医生..。我必须使用class_labels参数。

票数 0

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/52153288

复制

相似问题

问如何使用coreML转换为模型输出指定标签名称EN