OneCompiler

DM3

112

import numpy as np
import matplotlib.pyplot as plt
def estimate_coef(x, y):

n = np.size(x)
r
m_x = np.mean(x)
m_y = np.mean(y)

SS_xy = np.sum(yx) - nm_ym_x
SS_xx = np.sum(x
x) - nm_xm_x

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

plt.scatter(x, y, color = "m",
marker = "o", s = 30)

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

plt.plot(x, y_pred, color = "g")

plt.xlabel('x')
plt.ylabel('y')

plt.show()
def main():

x = np.array([0,1,2,3,4,5,6,7,8,9,11,13])
y = np.array([[1, 3, 2, 5, 7, 8, 8, 9, 10, 12,16, 18]])

b = estimate_coef(x, y)
print("Estimated coefficients:\nb_0 = {}
\nb_1 = {}".format(b[0], b[1]))
e
plot_regression_line(x, y, b)
if name == "main":
main(