线性拟合过程动态图

import matplotlib.pyplot as plt
import numpy as np
from sklearn import preprocessing
import matplotlib.font_manager as fm
data = np.loadtxt('./数据集/ex1data1.txt',delimiter=',')
plt.rcParams['font.sans-serif'] = ['SimHei']
x = data[:,:-1]
y = data[:,-1]
# xx = preprocessing.normalize(x,norm='l2')#归一缩放
X = np.c_[np.ones(len(x)),x]
print(X)
m,n = X.shape
# print(X)
# data = np.loadtxt('ex1data1.txt', delimiter=',')
# x = data[:, :-1]
# y = data[:, -1:]
# X = np.insert(x, 0, 1, axis=1)
# print(X)
plt.figure()
plt.ion()
def model(theta,X):
    h = X.dot(theta)
    return h
def costFun(h):
    J = 0.5*np.mean((h-y)**2)
    return J
def grad(alpha,iters):
    theta = np.zeros(n)
    J = np.zeros(iters)
    for i in range(iters):
        plt.title('动态变化图')
        plt.grid()

        h = model(theta,X)
        J[i] = costFun(h)
        detletheta = (1/m)*X.T.dot(h-y)
        theta -= alpha*detletheta
        plt.scatter(x, y, marker='.', c='r')
        plt.plot(x,h)
        plt.pause(0.1)
    plt.ioff()
    plt.show()
    return theta,J,h
theta,J,h = grad(0.01,2000)
print(theta)