''' Python实现曲线拟合 np.polyfit
评估拟合效果 np.polyval '''
import numpy as np
# b(x),a(y)是list或者numpy都可
loss = []
for num in range(1,8):
ff = np.polyfit(b,a,num)
a_ = np.polyval(ff,b)
MSE = [(a_[i]-a[i])**2 for i in range(len(a))]
MSE = np.mean(MSE)
loss.append(MSE)
print(f'num = {num},loss = {MSE}')
x = np.arange(1,8)
plt.plot(x,loss)
plt.scatter(x,loss,c='r')
plt.xlabel('n')
plt.ylabel('MSE(W)')
plt.show()
# num = 1,loss = 0.5295942624304562
# num = 2,loss = 0.47857519952795674
# num = 3,loss = 0.11569942797778665
# num = 4,loss = 0.07965323863079798
# num = 5,loss = 0.07609564489933388
# num = 6,loss = 0.05951952452568953
# num = 7,loss = 0.037572524325851515