Deep Learning

Multi-Class Classification? Yes.

Let’s discuss what is it!

Danyal Jamil
Dec 31, 2020 · 4 min read

Multi-class Classification

Softmax Regression

Image for post
Image for post
Model architecture for 4 predictions
Image for post
Image for post
same but labelled

Softmax Layer

Image for post
Image for post
Photo by Erik Mclean on Unsplash

Implementation

T = e^Z[l] # where Z[l] is the activation of the last layera[l] = (e^Z[l]) / (Sum(t[I]))a[l][I] = t[I] / (Sum(t[I]))

Loss Function

L(y, y`) = -Sum(y * log(y`))

Cost Function

J(hyper parameters) = (1 / m) * Sum(L(y`, y))

Conclusion

Contacts

Happy Learning. :)

Image for post
Image for post

The Startup

Medium's largest active publication, followed by +773K people. Follow to join our community.

Get the Medium app

A button that says 'Download on the App Store', and if clicked it will lead you to the iOS App store
A button that says 'Get it on, Google Play', and if clicked it will lead you to the Google Play store