Classification model
A model whose prediction is a class. For example, the following are all classification models:1
- A model that predicts an input sentence’s language (French? Spanish? Italian?).
- A model that predicts tree species (Maple? Oak? Baobab?).
- A model that predicts the positive or negative class for a particular medical condition.
Types of classification model
- Binary classification
- Multi-class classification
Metrics
- Use as a rough indicator of model training progress/convergence for balanced datasets.
- For model performance, use only in combination with other metrics.
- Avoid for imbalanced datasets. Consider using another metric.
- Use when false negative are more expensive than false positives.
- Use when false positives are more expensive than false negatives.
- Use when it’s very important for positive predictions to be accurate.
See also
- Regression models predict numbers rather than classes.
- ML crash course - Classification