Как использовать обученную модель, чтобы получить вероятность для каждого случая в Керасе

Question

Я сделал модель классификации изображений и хочу узнать точную вероятность каждого изображения перед нормализацией

как

☆ (введено изображение звезды)

-> {0,03, 0,07, 0,90} (результат для {p_square, p_circle, p_star})

Я использовал

output_array=model.predict_classes(test_image)

, он выводит только одно значение горячего кодирования, например

{1 , 0, 0}

Как вернуть вероятности прогноза?

Answer 1

вызов:

model.predict_proba(test_image)