import numpy as np
import matplotlib.pyplot as plt
x1 = np.array([0, 0, 1, 1])
x2 = np.array([0, 1, 0, 1])
y1 = np.array([0, 0, 0, 1])
y2 = np.array([0, 1, 1, 1]) # 真值表
alpha = 0.01
w0 = -0.8
w1 = 0.6
w2 = 0.4
xx = []
yy = []
zz = []
def fx(x1, x2):
return w1 * x1 + w2 * x2 + w0
def Fx(m, n):
if (fx(m, n) > 0):
return 1
else:
return 0
for i in range(5000):
w0 = w0 - (fx(x1[i % 4], x2[i % 4]) - y1[i % 4]) * alpha
w1 = w1 - (fx(x1[i % 4], x2[i % 4]) - y1[i % 4]) * x1[i % 4] * alpha
w2 = w2 - (fx(x1[i % 4], x2[i % 4]) - y1[i % 4]) * x2[i % 4] * alpha
xx.append(w0)
yy.append(w1)
zz.append(w2)
if (abs(fx(x1[i % 4], x2[i % 4]) - y1[i % 4]) < 0.001 and abs(
(fx(x1[i % 4], x2[i % 4]) - y1[i % 4]) * x1[i % 4]) < 0.001 and abs(
(fx(x1[i % 4], x2[i % 4]) - y1[i % 4]) * x2[i % 4]) < 0.01):
print(w0, w1, w2)
break
for i in range(5000):
w0 = w0 - (fx(x1[i % 4], x2[i % 4]) - y2[i % 4]) * alpha
w1 = w1 - (fx(x1[i % 4], x2[i % 4]) - y2[i % 4]) * x1[i % 4] * alpha
w2 = w2 - (fx(x1[i % 4], x2[i % 4]) - y2[i % 4]) * x2[i % 4] * alpha
xx.append(w0)
yy.append(w1)
zz.append(w2)
if (abs(fx(x1[i % 4], x2[i % 4]) - y2[i % 4]) < 0.001 and abs(
(fx(x1[i % 4], x2[i % 4]) - y2[i % 4]) * x1[i % 4]) < 0.001 and abs(
(fx(x1[i % 4], x2[i % 4]) - y2[i % 4]) * x2[i % 4]) < 0.01):
print(w0, w1, w2)
break
plt.plot(xx,"*")
plt.plot(yy,",")
plt.plot(zz,".")
plt.show()
运行结果: