numpy
Régression linéaire simple
Recherche…
Introduction
Adapter une ligne (ou une autre fonction) à un ensemble de points de données.
Utiliser np.polyfit
Nous créons un ensemble de données que nous ajustons ensuite avec une ligne droite $ f (x) = mx + c $.
npoints = 20
slope = 2
offset = 3
x = np.arange(npoints)
y = slope * x + offset + np.random.normal(size=npoints)
p = np.polyfit(x,y,1) # Last argument is degree of polynomial
Pour voir ce que nous avons fait:
import matplotlib.pyplot as plt
f = np.poly1d(p) # So we can call f(x)
fig = plt.figure()
ax = fig.add_subplot(111)
plt.plot(x, y, 'bo', label="Data")
plt.plot(x,f(x), 'b-',label="Polyfit")
plt.show()
Remarque: Cet exemple suit de très près la documentation numpy disponible sur https://docs.scipy.org/doc/numpy/reference/generated/numpy.polyfit.html .
Utiliser np.linalg.lstsq
Nous utilisons le même jeu de données qu'avec polyfit:
npoints = 20
slope = 2
offset = 3
x = np.arange(npoints)
y = slope * x + offset + np.random.normal(size=npoints)
Maintenant, nous essayons de trouver une solution en minimisant le système d'équations linéaires A b = c en minimisant | cA b | ** 2
import matplotlib.pyplot as plt # So we can plot the resulting fit
A = np.vstack([x,np.ones(npoints)]).T
m, c = np.linalg.lstsq(A, y)[0] # Don't care about residuals right now
fig = plt.figure()
ax = fig.add_subplot(111)
plt.plot(x, y, 'bo', label="Data")
plt.plot(x, m*x+c, 'r--',label="Least Squares")
plt.show()
Remarque: Cet exemple suit de près la documentation numpy à l' adresse https://docs.scipy.org/doc/numpy/reference/generated/numpy.linalg.lstsq.html .
Modified text is an extract of the original Stack Overflow Documentation
Sous licence CC BY-SA 3.0
Non affilié à Stack Overflow