File tree Expand file tree Collapse file tree 3 files changed +51
-0
lines changed
Data Analytics/Linear Regression Expand file tree Collapse file tree 3 files changed +51
-0
lines changed Original file line number Diff line number Diff line change
1
+ import pandas as pd
2
+ from matplotlib import pyplot as plt
3
+
4
+ def train (X ,y ):
5
+ mean_X = X .mean ()
6
+ mean_y = y .mean ()
7
+
8
+ numerator = 0
9
+ denominator = 0
10
+
11
+ for i in range (0 ,len (X )):
12
+ numerator += (X [i ] - mean_X )* (y [i ] - mean_y )
13
+ denominator += (X [i ] - mean_X )** 2
14
+
15
+ m = numerator / denominator
16
+ b = mean_y - (m * mean_X )
17
+
18
+ return m ,b
19
+
20
+ def plot (X ,y ,predicted ):
21
+ plt .xlabel ("Number of hours spent driving" )
22
+ plt .ylabel ("Risk Score" )
23
+ plt .scatter (X ,y )
24
+ plt .plot (X ,predicted ,color = "green" )
25
+ plt .savefig ("Ass1.jpg" )
26
+ plt .show ()
27
+
28
+
29
+ data = pd .read_csv ("data.csv" )
30
+ X = data .iloc [:,0 ]
31
+ y = data .iloc [:,1 ]
32
+
33
+ m ,b = train (X ,y )
34
+ predicted = []
35
+
36
+ for i in range (0 ,len (X )):
37
+ temp = (m * X [i ])+ b
38
+ predicted .append (temp )
39
+ print (temp ,y [i ])
40
+
Original file line number Diff line number Diff line change
1
+ X,Y
2
+ 8,3
3
+ 2,10
4
+ 11,3
5
+ 6,6
6
+ 5,8
7
+ 4,12
8
+ 12,1
9
+ 9,4
10
+ 6,9
11
+ 1,14
You can’t perform that action at this time.
0 commit comments