[go: up one dir, main page]
Scribd
Upload
48 views3 pages

Using A Dataset, Apply The Concept of Liner Regression

This document applies linear regression to a salary dataset to predict salary based on years of experience. It imports necessary packages, loads and splits the dataset, fits a linear regression model to the training data, makes predictions on the test data, and plots the results.

Uploaded by

19-361 Sai Prathik
Copyright
© © All Rights Reserved
We take content rights seriously. If you suspect this is your content, claim it here.
Available Formats
Download as DOC, PDF, TXT or read online on Scribd
48 views3 pages

Using A Dataset, Apply The Concept of Liner Regression

This document applies linear regression to a salary dataset to predict salary based on years of experience. It imports necessary packages, loads and splits the dataset, fits a linear regression model to the training data, makes predictions on the test data, and plots the results.

Uploaded by

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

EXPERIMENT- 2

// Using a dataset, apply the concept of Liner regression //


#import all the packages

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

# import the dataset

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

#   We are splitting our dataset so that we can train our model using 
a training dataset 
#  test the model using a test dataset.
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)

#  used a fit() method to fit our Simple Linear Regression object to t
he training set. 
from sklearn.linear_model import LinearRegression
regressor = LinearRegression()
regressor.fit(X_train, y_train)

# We will create a prediction vector y_pred, and x_pred, which wi
ll contain predictions of test dataset, and prediction of trainin
g set respectively.
y_pred = regressor.predict(X_test)

# Train the model

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')
plt.ylabel('Salary')
plt.show()

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()

OUTPUT:

You might also like