AIExplainer
Machine Learning Intermediate 2 min read

What is generalization curve?

A graph that shows how well a machine learning model performs on new, unseen data as the amount of training data increases

The generalization curve is a tool used to evaluate the performance of a machine learning model. It plots the model's performance on a test set against the amount of training data used to train the model. The curve helps to identify whether the model is overfitting or underfitting the data.

Think of the generalization curve like a student's learning curve. At first, the student may struggle to understand a new concept, but as they practice and receive more instruction, their performance improves. However, if the student practices too much, they may start to memorize rather than truly understand the material, and their performance may plateau or even decline. Similarly, a machine learning model's performance may improve with more training data, but eventually, it may start to overfit and its performance on new data may decline.

For example, a company building a machine learning model to predict customer churn might use a generalization curve to determine how much training data is needed to achieve a certain level of accuracy. If the curve shows that the model's performance improves significantly with more training data, the company may decide to collect more data to improve the model's accuracy.

The generalization curve is used to diagnose issues with a machine learning model's performance, such as overfitting or underfitting. It can also be used to determine the optimal amount of training data needed to achieve good performance.

One common misconception is that more training data is always better. However, the generalization curve can show that too much training data can lead to overfitting, which can actually decrease the model's performance on new data.

The concept of the generalization curve has been around since the early days of machine learning, but it has become increasingly important as machine learning models have become more complex and widely used.

learning curve performance curve

Three products for different needs — explore what’s relevant to you.