Machine Learning Project 1: Predict Salary using Simple Linear Regression
I’m going to be writing a series of articles starting from basic Machine Learning projects all the way to advanced ones.
#100DaysOfMLCode #100ProjectsInML
The best approach for me to learn anything is by working on sample projects. No matter how simple the project is, it helps me better understand the concepts. So I will be working through some small mini projects as part of this learning journey.
There are 100’s of excellent resources out there to help you get started. I stumbled upon this A-Z Machine learning course on Udemy and I’ll be walking through those examples in the first few weeks.
Today I’ll be going through “Simple Linear Regression”
Dataset
First let’s look at the dataset. It is Salary_Data.csv and can be found here.
It has 2 columns — “Years of Experience” and “Salary” for 30 employees in a company. So in this example, we will train a Simple Linear Regression model to learn the correlation between the number of years of experience of each employee and their respective salary. Once the model is trained, we…