Я обучил модель CNN, которая является классификатором мультикласса.Я использовал единицу softmax в выходном слое, но сейчас я не знаю класс, связанный с отдельными вероятностями.
Я использую model.predict_proba, чтобы получить матрицу вероятностей
class_mode = 'категорический', classes = ["zipper", "backstrap", "slip_on", "lace_up"])
Если выходные данные model.predict_proba = [0.2,0.2,0.2,0.4] Какой класс соответствует 0,4?
Я думаю, что это в алфавитном порядке.Backstrap = 0, lace_up = 1, slip_on = 2, молния = 3, но я не уверен