python - sklearnで多項式回帰を行う最も簡単な方法は?

Question

線形回帰に適合しないデータがいくつかあります。

実際、二次関数に「正確に」適合する必要があります。

P = R*I**2 

私はこれを作っています：

model = sklearn.linear_model.LinearRegression()

X = alambres[alambre]['mediciones'][x].reshape(-1, 1)
Y = alambres[alambre]['mediciones'][y].reshape(-1, 1)
model.fit(X,Y)

次のようなことで解決する可能性はありますか:

model.fit([X,X**2],Y)

Answer 1

numpy のpolyfitを使用できます。

import numpy as np
from matplotlib import pyplot as plt
X = np.linspace(0, 100, 50)
Y = 23.24 + 2.2*X + 0.24*(X**2) + 10*np.random.randn(50) #added some noise
coefs = np.polyfit(X, Y, 2)
print(coefs)
p = np.poly1d(coefs)
plt.plot(X, Y, "bo", markersize= 2)
plt.plot(X, p(X), "r-") #p(X) evaluates the polynomial at X
plt.show()

外：

[  0.24052058   2.1426103   25.59437789]