随机梯度下降

代码
import random
import dataset
import matplotlib.pyplot as plt
xs, ys = dataset.get_beans(100)

plt.title(“Size-Toxicity Function”, fontsize=12)
plt.xlabel(“Size”)
plt.ylabel(“Toxicity”)
plt.scatter(xs, ys)
w = 0.1
y_pre = w * xs
plt.plot(xs, y_pre)
plt.show()

for _ in range(100):
i = random.randint(0, xs.size-1)
x = xs[i]
y = ys[i]
k = 2 * (x ** 2) * w + (-2 * x * y)
alpha = 0.1
w = w - alpha * k
plt.clf()
plt.scatter(xs, ys)
y_pre = w * xs
plt.xlim(0, 1)
plt.ylim(0, 1.2)
plt.plot(xs, y_pre)
plt.pause(0.01)