随机梯度下降

import numpy as np
import math
import random

theta0 = np.random.random()
theta1 = np.random.random()
theta2 = np.random.random()
alpha = 0.01
x1 = np.array([2104,1600, 2400,1416,3000])
x2 = np.array([3,3,3,2,4])
t = np.array([400,330,369,232,540])
eps = math.exp(-10)
e0 = 0.00001
e1 = 0.00001
e2 = 0.00001

while e0 >= eps or e1 >=eps or e2 >= eps:
    i = random.randint(0,4)
    e0 = theta0 + theta1*x1[i] +theta2*x2[i] - t[i]
    e1 = (theta0 + theta1*x1[i] +theta2*x2[i] - t[i])*x1[i]    
    e2 = (theta0 + theta1*x1[i] +theta2*x2[i] - t[i])*x2[i]
    theta0 -= alpha*e0
    theta1 -= alpha*e1
    theta2 -= alpha*e2
        
print(theta0,theta1,theta2)

0.19211389060616124 0.6215045558737046 0.3111396512322576