От вывода softmax до прогноза класса

Question

Есть ли простой способ go из вывода Softmax к предсказанию класса?

Например,

из этого:

[0.83128697, 0.06161868, 0.10709436]

к этому :

[1, 0, 0]

Answer 1

Вы можете использовать np.argmax для получения индекса максимального значения:

import numpy as np
a = [0.83128697, 0.06161868, 0.10709436]

r = np.zeros(len(a)) # a.size if a is a numpy array
r[np.argmax(a)]=1

r
array([1., 0., 0.])