What is a learning curve
To produce a learning curve, many models are trained using a different proportion of the data available for training. For each proportion, the model performance metric (mean square error in our case) will be calculated. In some cases, to avoid having too much variability in the results, it is possible to train multiple models for the same proportion of data and get an average (see curve on test data on the left). On the Figure shown below for instance, you can see that training a model with more data will improve the model’s performance as the mean square error (MSE) keeps going down for the test data. Point 1 shows the MSE when the model is trained on 50% of the available data, and point 2 shows the MSE when the model is trained on 100% of the available data.
How to read a learning curve?
In the examples below, we trained 3 models (linear regression, polynomial regression and decision tree regression) on the same data and we plotted the learning curve for each model:![]() | Example for an underfitting model: Linear regression
|
![]() | Example for a suitable model: Polynomial regression
|
![]() | Example for an overfitting model: Decision Tree Regression
|


