aNet.ai
Machine Learning/Learning path
0day streak0XP · Lv 1
← 7. Classification
Lesson 1 of 6 7 min · 155 words

From regression to classification

Why a straight line is a poor model for yes/no targets.

In classification the target is a category, encoded for binary problems as . You could fit ordinary linear regression to 0/1 targets and threshold at 0.5 — but it works poorly.

The fitted line produces values below 0 and above 1, which can't be probabilities. Worse, a few points far from the boundary pull the line around and shift the threshold, even though they are already easy to classify. Squared error also punishes predictions that are 'too correct' (e.g. 1.8 for a positive).

The fix: keep a linear score , but squash it into with an S-shaped function and train it with a loss suited to probabilities. That is logistic regression.

Formula to remember

Sigmoid

Maps any real number into (0, 1).

Quick check · +5 XP

σ(0) equals…

Practice this topic
Saved in this browser as you type.
aNet.ai · Introduction to Machine Learning · NewUU course outlineProgress saved in this browser