Jaká je role softmax v modelu a k čemu se používá?

Softmax je aktivační funkce, která přeměňuje vektor libovolných čísel na vektor pravděpodobností, jejichž součet je roven 1. V modelech strojového učení, zejména v úlohách klasifikace, se softmax používá na výstupní vrstvě neuronové sítě k získání pravděpodobností příslušnosti ke každé třídě. To umožňuje interpretovat výstup modelu jako rozdělení pravděpodobností a vybrat třídu s nejvyšší pravděpodobností.

Příklad: pokud jsou na výstupu modelu hodnoty [2.0, 1.0, 0.1], aplikace softmax je přemění přibližně na [0.65, 0.24, 0.11].