Как получить метку максимального значения при использовании функции активации softmax в выходном слое нейронной сети?

Question

в модели, которую я обучил, я применяю функцию softmax в выходном слое нейронной сети.вывод имеет 41 категорию, и я хочу получить метку с максимальным значением и само значение ..i.е в выводе.41 болезнь для набора входных данных .... Softmax предсказывает для всех болезней, но я хочу напечатать болезнь с максимальной вероятностью наряду с вероятностью, как мне это сделать?

Answer 1

Этого можно добиться, просто используя функцию np.argmax():

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

predictions = model.predict(x_test)
print(np.argmax(predictions[0])  #output the index of the disease with max proba
                                 #for example N#0