梯度下降法拟合曲线

import random

import numpy as np

x=[[2104,3],[1600,3],[2400,3],[1416,2],[3000,4]]
t=[400,330,369,232,540]
n=len(x)
c=[]
h=[]
ti=[]
y=[]
e=[]
alpha=0.01
for i in range(3):
    c.append(random.random())
for i in range(5):
    x[i].append(1)
for i in range(5):
    h.append(sum(np.multiply(c,x[i])))
for i in range(5):
    # ti.append(np.multiply((h[i]-t[i]),x[i]))
    y.append(h[i]-t[i])
print(y)
flag = True
b1=[a[0] for a in x]
b2=[a[1] for a in x]
print(alpha*sum(np.multiply(y,b1)))
while 1:
    if flag:
        i=0
        for i in range(5):
            if (y[i]<=0.0001):
                flag=False
            # for j in range(5):
            #     for k in range(2):
            #         c[k]=c[k]+alpha*(y[j])*x[j,k]
            c[1]=c[1]-alpha*sum(np.multiply(y,b1))
            c[2] = c[2] - alpha * sum(np.multiply(y, [a[1] for a in x]))
            for j in range(5):
                h=sum(np.multiply(c, x[j]))
                y[j]=h-t[j]
    else:
        break

print(c)

[559.8932283591981, 400.59021166245236, 725.5632540382394, 414.03443016849974, 828.3850175676012]
66317.3930664085
[0.45496630296973384, -789707062695191.6, -1119850464843.8948]