Поиск полиномиального уравнения - PullRequest
Новая
       29

Поиск полиномиального уравнения

0 голосов
/ 10 июня 2018

Существует набор данных с x и y, и мне нужно знать уравнение для y (как оно было смоделировано). Поэтому я использую коэффициенты и мощность из модели с 99% -ным счетом: 

poly = PolynomialFeatures(1)
Xtrain_tr = poly.fit_transform(Xtrain)
Xtest_tr = poly.fit_transform(Xtest)
model = LinearRegression(fit_intercept=True)
model.fit(Xtrain_tr, ytrain)
poly=poly.powers_
coef=model.coef_
coef
out:    array([  0.        ,   4.25571706,   0.59691212,  20.28394618,
           162.97671755])

    poly
 out:   array([[0, 0, 0, 0],
           [1, 0, 0, 0],
           [0, 1, 0, 0],
           [0, 0, 1, 0],
           [0, 0, 0, 1]], dtype=int64)

И здесь я пытаюсь имитировать значение y для первой строки в наборе данных: 

temp=0
final=0
i=0

for i in range(4):
    temp=(X_train[:1]**poly[i])*coef[i]
    temp=temp.values.sum()

    final=final+temp
    temp=0
    i=i+1

Результат неудовлетворительный, пожалуйста, предложите, что я могу улучшить здесь.Спасибо,

...