我应该如何使用 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 参数。