0% found this document useful (0 votes)
59 views5 pages

Linear Regression Implementation

The document describes implementing linear regression on a dataset. It defines functions to estimate coefficients, plot the regression line, and contains code to call the functions on a sample x and y dataset.

Uploaded by

Avnish Patel
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
0% found this document useful (0 votes)
59 views5 pages

Linear Regression Implementation

The document describes implementing linear regression on a dataset. It defines functions to estimate coefficients, plot the regression line, and contains code to call the functions on a sample x and y dataset.

Uploaded by

Avnish Patel
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

Experiment 1.

Student Name: Avnish Kumar UID: 20BCS3331


Branch: CSE Section/Group: 614-A
Semester: 5 Subject Name: Machine Learning Lab

Aim: Implement linear regression on any dataset.

Code:
import numpy as np
import [Link] as plt

def estimate_coef(x, y):


# number of observations/points
n = [Link](x)

# mean of x and y
vector m_x =
[Link](x) m_y =
[Link](y)

# calculating cross-deviation and deviation about x


SS_xy = [Link](y*x) - n*m_y*m_x
SS_xx = [Link](x*x) - n*m_x*m_x

# calculating regression coefficients


b_1 = SS_xy / SS_xx
b_0 = m_y - b_1*m_x

return (b_0, b_1)

def plot_regression_line(x, y, b):


# plotting the actual points as scatter plot
[Link](x, y, color = "m", marker = "o", s = 30)

# predicted response vector


y_pred = b[0] + b[1]*x

# plotting the regression line


[Link](x, y_pred, color = "g")

# putting labels
[Link]('x')
[Link]('y')

# function to show plot


[Link]()

def main():
# observations / data
x = [Link]([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
y = [Link]([1, 3, 2, 5, 7, 8, 8, 9, 10, 12])

# estimating coefficients b = estimate_coef(x, y)


print("Estimated coefficients:\nb_0 = {} \\nb_1 = {}".format(b[0],
b[1]))

# plotting regression line


plot_regression_line(x, y, b)

if name == " main ":


main()
Output

You might also like