Swipe
SwipeSwipe
Swipe Apps

Paths

Motivation

Information

Classification vs. Regression · Lesson 5 of 9

Classification: Picking a Category

A free lesson from Swipe AI · Machine Learning Fundamentals, level 2 · updated October 2026

By the Swipe team, founded by Colin · how we make lessons

Classification isn't limited to two categories. Binary classification picks between exactly two options, like spam or not spam, but multi-class classification sorts an input into one of many possible categories — like tagging a photo as 'cat,' 'dog,' 'bird,' or 'other.'

Under the hood, most classifiers don't just blurt out one answer — they estimate how likely the input belongs to each category, then pick whichever category comes out most likely. A spam filter might be 92% confident an email is spam and 8% confident it isn't; it reports 'spam' because that's the stronger signal.

The stakes of choosing the right category list show up clearly in something like a medical scan triage tool, which might sort scans into 'urgent review needed,' 'routine follow-up,' or 'no action needed.' If that category list is missing an important case, the model has no way to express it — it can only pick from what it was given.

Input: photo of an animal
Model's category options: cat, dog, bird, other
Model's confidence: cat 71%, dog 12%, bird 4%, other 13%
Predicted category: cat (highest confidence)
How a multi-class classifier picks its final answer
Quick check

A photo-tagging tool sorts every image into "landscape," "portrait," "food," or "other." What kind of task is this?

Fill in the blank

A classifier that estimates how likely an input belongs to each category, and returns the category with the highest likelihood, is choosing based on ____.

True or false

A classification model can only choose from the fixed list of categories it was trained on, even if the true answer falls outside that list.

Keep your streak going in Swipe AI

Swipe through every level of Machine Learning Fundamentals on your iPhone — with XP, stars, streaks, a personal advisor and the full track beyond these free levels.

Get the Swipe apps →