Machine Learning

Softmax

A function that converts a vector of numbers into a probability distribution, where each value is between 0 and 1 and all values sum to 1. It is typically used as the final layer in classification models.

Why It Matters

Softmax gives you not just a prediction but a confidence level for each possible class, enabling better decision-making in applications like medical diagnosis.

Example

A model outputs raw scores [2.0, 1.0, 0.1] for three classes. Softmax converts these to probabilities [0.66, 0.24, 0.10], showing 66% confidence in class 1.

Think of it like...

Like converting test scores into percentages of the total — it shows each option's relative likelihood compared to all others.

Related Terms