有一个带有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
结果不满意,请在此提出我可以改进的建议。 谢谢,