How does the regression model work?

Our objective is to train a machine learning model on the dataset and then ask the model to make predictions in order to establish the salary that should be given to an employee based on their years of experience.

The example that we are considering is based on an Excel sheet. Basically, we have data from a company where we have a salary structure based on years of experience. We want our machine learning model to derive the correlation between the years of experience and the salary given. From the derived correlation, we want the model to provide future predictions and specify the modeled salary. The machine does this through simple linear regression. In simple linear regression, various lines are drawn ...

Get Hands-On Penetration Testing with Python now with the O’Reilly learning platform.

O’Reilly members experience books, live events, courses curated by job role, and more from O’Reilly and nearly 200 top publishers.