Scribd
Upload
47 views

Python Code To Implement Linear Regression

This document describes Python code to implement linear regression. It includes sample code to fit a linear regression model to input data, output a scatter plot of the data with the best fit regression line, and predict future y values based on new x values. The code is applied to sample dataset of years of experience and salary for employees to predict salary based on years of experience.

Uploaded by

Dreaming Boy
Copyright
© © All Rights Reserved
Available Formats
Download as DOCX, PDF, TXT or read online on Scribd
47 views

Python Code To Implement Linear Regression

This document describes Python code to implement linear regression. It includes sample code to fit a linear regression model to input data, output a scatter plot of the data with the best fit regression line, and predict future y values based on new x values. The code is applied to sample dataset of years of experience and salary for employees to predict salary based on years of experience.

Uploaded by

Dreaming Boy
Copyright
© © All Rights Reserved
Available Formats
Download as DOCX, PDF, TXT or read online on Scribd
You are on page 1/ 5

Python Code to implement Linear Regression

Input a Dataset and the X value to predict future Y.

Apply Regression Algorithm

Output :

Scatter Plot and Best Regression Line.

and Predicted Y Value

Code:
class linregr():
  def __init__(self):
    self.coeff1=0
    self.coeff0=0
  def fitting(self,x,y):
    def mean(val):
      return sum(val)/len(val)
    xm=mean(x)
    ym=mean(y)
    xx=[i-xm for i in x]
    yy=[i-ym for i in y]
    xx2=[i*i for i in xx]
    xxyy=[xx[i]*yy[i] for i in range(len(xx))]
    self.coeff1=sum(xxyy)/sum(xx2)
    self.coeff0=ym-(self.coeff1*xm)
  def predict(self,args):
    li=[]
    if(type(args)==type([])):
      pass
    else:
      arg=[]
      arg.append(args)
      args=arg
    for i in args:
      li.append((self.coeff1*float(i))+self.coeff0)
    return li
Dataset:
YearsExperi Sala
ence ry
2508
1.6 1
2665
1.8 9
4509
2.1 7
4783
2.4 3
5755
2.7 0
5850
2.8 7
6201
3.2 4
6307
3.4 6
6770
3.8 1
7298
4.3 8
7598
5.1 7
7805
5.4 0
8090
5.5 9
9310
5.8 3
9473
6.3 7
1009
6.4 01
1019
7.1 91
1045
7.7 00
1073
8.1 84
1085
8.2 35
1107
8.4 31
1192
8.6 43
1201
9.7 22
1233
9.9 30
1239
10.1 61
1262
10.2 72
1286
10.2 25
1292
10.7 43
1356
10.7 01
1394
10.9 63

Output :
Colab Notebook Link:

https://colab.research.google.com/drive/1HiyvAoQqsZtfzmW
4pdlek_wcVvWA6OSB?usp=sharing

You might also like