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)
线性拟合过程动态图
最新推荐文章于 2024-04-29 15:01:12 发布