Как мне дать имена меток для вывода модели с помощью 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.