How to plot Learning Curve in Python

Hits: 998

How to plot Learning Curve in Python

When building a machine learning model, it’s important to evaluate its performance and make sure it’s not overfitting or underfitting. One way to do this is by plotting a learning curve.

A learning curve is a graph that shows the relationship between the model’s performance and the amount of data it’s trained on. It helps to understand how the model’s performance improves as more data is added.

In Python, the library scikit-learn provides an easy way to plot a learning curve using the function learning_curve().

The first step is to import the library and load the dataset into a pandas dataframe. Then, split the data into training and testing sets, and create an instance of the model you want to evaluate.

After that, use the learning_curve() function, which takes the model, the dataset, and the number of folds as inputs. The function will return the train and test scores for different training set sizes.

You can use the ‘train_sizes’ parameter to set different training set sizes, by default it uses np.linspace(0.1, 1.0, 5) that means the data is divided into 5 sets.

Finally, you can use the matplotlib library to plot the scores, by plotting the training and testing scores separately, you can get an idea of the model’s performance with different data size.

In summary, a learning curve is a powerful tool for evaluating the performance of a machine learning model and make sure it’s not overfitting or underfitting. By using the learning_curve() function in scikit-learn, it’s easy to plot a learning curve in Python, making it a valuable tool for data scientists and machine learning practitioners.


In this Learn through Codes example, you will learn: How to plot Learning Curve in Python.


Personal Career & Learning Guide for Data Analyst, Data Engineer and Data Scientist

Applied Machine Learning & Data Science Projects and Coding Recipes for Beginners

A list of FREE programming examples together with eTutorials & eBooks @ SETScholars

95% Discount on “Projects & Recipes, tutorials, ebooks”

Projects and Coding Recipes, eTutorials and eBooks: The best All-in-One resources for Data Analyst, Data Scientist, Machine Learning Engineer and Software Developer

Topics included: Classification, Clustering, Regression, Forecasting, Algorithms, Data Structures, Data Analytics & Data Science, Deep Learning, Machine Learning, Programming Languages and Software Tools & Packages.
(Discount is valid for limited time only)

Disclaimer: The information and code presented within this recipe/tutorial is only for educational and coaching purposes for beginners and developers. Anyone can practice and apply the recipe/tutorial presented here, but the reader is taking full responsibility for his/her actions. The author (content curator) of this recipe (code / program) has made every effort to ensure the accuracy of the information was correct at time of publication. The author (content curator) does not assume and hereby disclaims any liability to any party for any loss, damage, or disruption caused by errors or omissions, whether such errors or omissions result from accident, negligence, or any other cause. The information presented here could also be found in public knowledge domains.

Learn by Coding: v-Tutorials on Applied Machine Learning and Data Science for Beginners


Machine Learning for Beginners in Python: How to Plot The Learning Curve

How to plot Validation Curve in Python