Aká je úloha softmax v modeli a na čo sa používa?

Softmax je aktivačná funkcia, ktorá premení vektor ľubovoľných čísel na vektor pravdepodobností, ktorých súčet je rovný 1. V modeloch strojového učenia, najmä pri úlohách klasifikácie, sa softmax používa na výstupnej vrstve neurónovej siete na získanie pravdepodobností príslušnosti ku každej triede. To umožňuje interpretovať výstup modelu ako rozdelenie pravdepodobností a vybrať triedu s najvyššou pravdepodobnosťou.

Príklad: ak výstup modelu obsahuje hodnoty [2.0, 1.0, 0.1], aplikácia softmax ich premení na približne [0.65, 0.24, 0.11].