[云炬python3玩转机器学习] 5-3简单线性回归的实现

 

03 实现简单线性回归法

In [20]:

import numpy as np import matplotlib.pyplot as plt import datetime print(datetime.datetime.now()) 2022-01-17 20:41:47.177206

In [21]:

x = np.array([1., 2., 3., 4., 5.]) y = np.array([1., 3., 2., 3., 5.])

In [38]:

plt.scatter(x, y) plt.axis([0, 6, 0, 6]) plt.show()

In [23]:

x_mean = np.mean(x) y_mean = np.mean(y)

In [24]:

num = 0.0 d = 0.0 for x_i, y_i in zip(x, y):num += (x_i - x_mean) * (y_i - y_mean)d += (x_i - x_mean) ** 2

In [28]:

a = num/d a

Out[28]:

0.8

In [29]:

b = y_mean - a * x_mean b

Out[29]:

0.39999999999999947

In [30]:

y_hat = a * x + b

In [31]:

plt.scatter(x, y) plt.plot(x, y_hat, color='r') plt.axis([0, 6, 0, 6]) plt.show()

In [32]:

x_predict = 6 y_predict = a * x_predict + b y_predict

Out[32]:

5.2

总结

以上是生活随笔为你收集整理的[云炬python3玩转机器学习] 5-3简单线性回归的实现的全部内容，希望文章能够帮你解决所遇到的问题。

如果觉得生活随笔网站内容还不错，欢迎将生活随笔推荐给好友。