状态模型预测列车数据:未对齐的形状

Question

import numpy as np
import statsmodels.api as sm
duncan_prestige = sm.datasets.get_rdataset("Duncan", "carData")
Y = duncan_prestige.data['income']
print(Y.shape)
X = duncan_prestige.data['education']
print(X.shape)
X = sm.add_constant(X)
print(X.shape)
model = sm.OLS(Y,X)
results = model.fit()
model.predict(X)
results.params

错误: ValueError:形状( 45 ,2)和(45,2)不对齐:2 (dim 1) != 45 (dim 0)。如何预测训练集？(我需要通过plt.plot绘制带有更多数据的回归线)

Answer 1

应该打电话来

results.predict(X)