ML Learning Hub
Classificationintermediate

OvA vs OvO Multi-class Classification

Extending binary classifiers to multi-class — tournament brackets for algorithms

One-vs-All and One-vs-One strategies for extending binary classifiers to multi-class — decision boundaries, scalability, SVM applications, and when to use Softmax instead.

30 min
6 diagrams
6 Concepts Covered

Prerequisites

SVM
Logistic Regression

Concepts Covered

Multi-classDecision BoundariesOvAOvOSoftmaxClass Imbalance

Key Formulas

OvA Classifiers

K binary classifiers, one per class vs. all others

OvO Classifiers

One binary classifier per pair of classes

Softmax

Normalizes K logits to a probability distribution

Interactive Simulation

Loading visualization…

Model Architecture

Loading visualization…
🎯

The Multi-Class Problem

motivation

Many real problems have more than 2 classes: digit recognition (10 classes), species classification (100s), product categorization (1000s). Some algorithms (logistic regression, SVMs) are inherently binary. Two strategies extend them: OvA trains K classifiers, each separating class k from all others. OvO trains K(K-1)/2 classifiers for every pair. Neural networks with Softmax solve multi-class natively.

⚖️

OvA vs OvO vs Softmax

comparison
1

OvA: K classifiers, each uses all data. Fast training. Imbalanced (1 positive vs K-1 negatives). Good for large K.

2

OvO: K(K-1)/2 classifiers, each uses only 2 classes. Balanced but slow for large K (100 classes = 4950 classifiers).

3

Softmax (multinomial LR): single model, K outputs, trained with cross-entropy. Most efficient. Native to neural nets.

4

SVM convention: OvO is default in sklearn (historically performs slightly better). For neural nets, always Softmax.

</>

Softmax Multi-class Classification

code
python49 lines
import torch
import torch.nn as nn
from sklearn.datasets import make_classification
from sklearn.model_selection import train_test_split
from sklearn.multiclass import OneVsRestClassifier, OneVsOneClassifier
from sklearn.svm import SVC

# ── Sample data ────────────────────────────────────────────────────────
X_np, y_np = make_classification(n_samples=300, n_features=8,
                                  n_classes=3, n_informative=6, random_state=42)
X_train_np, X_test_np, y_train_np, _ = train_test_split(
    X_np, y_np, test_size=0.2, random_state=42)

# ── PyTorch multiclass setup ───────────────────────────────────────────
K = 3                                       # number of classes
batch = 16

# Tiny 2-layer net for the demo
class SimpleNet(nn.Module):
    def __init__(self): super().__init__(); self.fc = nn.Linear(8, K)
    def forward(self, x): return self.fc(x)

model = SimpleNet()
x = torch.randn(batch, 8)                   # one mini-batch
y = torch.randint(0, K, (batch,))           # class indices

# Class weights (handle imbalance)
class_weights = torch.tensor([1.0, 2.0, 1.5])   # weight rarer classes higher

# Softmax + Cross-Entropy (combined for numerical stability)
criterion = nn.CrossEntropyLoss(
    weight=class_weights,    # For imbalanced classes
    label_smoothing=0.1      # Prevents overconfident predictions
)

# Model outputs raw logits (no softmax in forward pass)
logits = model(x)            # Shape: (batch, K)
loss = criterion(logits, y)  # y contains class indices
print(f"Multiclass CE loss: {loss.item():.4f}")

# Predictions
probs = torch.softmax(logits, dim=-1)
preds = probs.argmax(dim=-1)

# Sklearn: OvR (OvA) strategy
ovr = OneVsRestClassifier(SVC(kernel='rbf', probability=True))
ovo = OneVsOneClassifier(SVC(kernel='rbf'))
ovr.fit(X_train_np, y_train_np)
print(f"OvR accuracy: {ovr.score(X_test_np, _):.3f}")

?Knowledge Check

Progress is saved in your browser — no account needed.

Need a Data Scientist or AI Engineer?

I build custom ML models, RAG chatbots, data pipelines, and production APIs — from analysis to deployment.