Scribd
Upload
13 views3 pages

Simple Linear Regression Code

The document outlines the process of implementing Simple Linear Regression using Python, including importing necessary libraries, loading a dataset, and splitting it into training and test sets. It details training the model, making predictions, and visualizing results for both training and test sets. Additionally, it provides the final regression equation and explains how to interpret the coefficients and intercept obtained from the model.

Uploaded by

shalini gambhir
Copyright
© © All Rights Reserved
We take content rights seriously. If you suspect this is your content, claim it here.
Available Formats
Download as PDF, TXT or read online on Scribd
13 views3 pages

Simple Linear Regression Code

The document outlines the process of implementing Simple Linear Regression using Python, including importing necessary libraries, loading a dataset, and splitting it into training and test sets. It details training the model, making predictions, and visualizing results for both training and test sets. Additionally, it provides the final regression equation and explains how to interpret the coefficients and intercept obtained from the model.

Uploaded by

shalini gambhir
Copyright
© © All Rights Reserved
We take content rights seriously. If you suspect this is your content, claim it here.
Available Formats
Download as PDF, TXT or read online on Scribd
You are on page 1/ 3

Dr.

Shalini Gambhir
AIML-Fundamentals of Machine Learning

Simple Linear Regression

Importing the libraries

import numpy as np
import matplotlib.pyplot as plt
import pandas as pd

Importing the dataset

dataset = pd.read_csv('Salary_Data.csv')
X = dataset.iloc[:, :-1].values
y = dataset.iloc[:, -1].values

Splitting the dataset into the Training set and Test set

from sklearn.model_selection import train_test_split


X_train, X_test, y_train, y_test = train_test_split(X, y, test_size = 1/3, random_state = 0)

Training the Simple Linear Regression model on the Training set

from sklearn.linear_model import LinearRegression


regressor = LinearRegression()
regressor.fit(X_train, y_train)

LinearRegression(copy_X=True, fit_intercept=True, n_jobs=None, normalize=False)

Predicting the Test set results

y_pred = regressor.predict(X_test)

Visualising the Training set results

plt.scatter(X_train, y_train, color = 'red')


plt.plot(X_train, regressor.predict(X_train), color = 'blue')
plt.title('Salary vs Experience (Training set)')
plt.xlabel('Years of Experience')

Dr. Shalini Gambhir


AIML-Fundamentals of Machine Learning
Dr. Shalini Gambhir
AIML-Fundamentals of Machine Learning

plt.ylabel('Salary')
plt.show()

Visualising the Test set results

plt.scatter(X_test, y_test, color = 'red')

plt.plot(X_train, regressor.predict(X_train), color = 'blue')

plt.title('Salary vs Experience (Test set)')

plt.xlabel('Years of Experience')

plt.ylabel('Salary')

plt.show()

Making a single prediction (for example the salary of an employee with 12 years of experience)

print(regressor.predict([[12]]))

[138967.5015615]

Therefore, our model predicts that the salary of an employee with 12 years of experience is $
138967,5.

Notice that the value of the feature (12 years) was input in a double pair of square brackets. That's
because the "predict" method always expects a 2D array as the format of its inputs. And putting 12
into a double pair of square brackets makes the input exactly a 2D array. Simply put:

12→scalar

[12]→1D array

[[12]]→2D array

Getting the final linear regression equation with the values of the coefficients

print(regressor.coef_)
print(regressor.intercept_)

[9345.94244312]

26816.192244031183

Therefore, the equation of our simple linear regression model is:

Salary=9345.94×YearsExperience+26816.19

Dr. Shalini Gambhir


AIML-Fundamentals of Machine Learning
Dr. Shalini Gambhir
AIML-Fundamentals of Machine Learning

To get these coefficients we called the "coef_" and "intercept_" attributes from our regressor object.
Attributes in Python are different than methods and usually return a simple value or an array of
values.

Dr. Shalini Gambhir


AIML-Fundamentals of Machine Learning

You might also like