如何使用coreML convert将标签名称赋予模型输出

Question

我已经使用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？

Answer 1

没关系...我只是没有正确阅读文档。.我不得不使用class_labels参数。