提示:本站收集StackOverFlow近2千万问答，支持中英文搜索，鼠标放在语句上弹窗显示对应的参考中文或英文， 本站还提供 中文繁体 英文版本 中英对照 版本，有任何建议请联系yoyou2525@163.com。
我正在尝试编写一个 Python 程序来生成使用随机变量（具有高斯分布）和四次多项式方程（3x^4+x^3+3x^2+4x+5）之和的数据. 使用最小二乘多项式拟合，使用模型对生成的数据进行曲线化，直到您的模型可以准确预测所有值。 我是 python 的新手，真的很想赶上我的快节奏课程。 任何帮助和进一步的解释将不胜感激。 我在没有 for 循环的情况下尝试了它，它给出了两条曲线，但我认为它应该匹配初始点。 请看下面的代码：
import random
import matplotlib.pyplot as plt
import numpy as np
def rSquared(obs, predicted):
error = ((predicted - obs)**2).sum()
mean = error/len(obs)
return 1 - (mean/np.var(obs))
def generateData(a, b, c, d, e, xvals):
for x in xvals:
calcVal= a*x**4 + b*x**3 + c*x**2 + d*x + e
yvals.append(calcVal+ random.gauss(0, 35))
xvals = np.arange(-10, 11, 1)
yvals= []
a, b, c, d, e = 3, 1, 3, 4, 5
generateData(a, b, c, d, e, xvals)
for i in range (5):
model= np.polyfit(xvals, yvals, i)
estYvals = np.polyval(model, xvals)
print('R-Squared:', rSquared(yvals, estYvals))
plt.plot(xvals, yvals, 'r', label = 'Actual values')
plt.plot(xvals, estYvals, 'bo', label = 'Predicted values')
plt.xlabel('Variable x values')
plt.ylabel('Calculated Value of Polynomial')
plt.legend()
plt.show()
是的，你只需要尝试
model = np.polyfit(xvals, yvals, i) # i=4 to get the perfect fit to values with R-Square of 4 : 0.9999995005089268.
声明：本站的技术帖子网页，遵循CC BY-SA 4.0协议，如果您需要转载，请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.