我有一个 dataframe
,我试图运行 statsmodel.api
OLS回归 . 它打印出摘要 . 但是当我使用 predict()
函数时,它给了我一个错误 -
形状(75,7)和(6,)未对齐:7(暗淡1)!= 6(暗淡0)
我的代码是:
X = newdf.loc[:, newdf.columns != 'V-9'].values
y = newdf.iloc[:,3].values
from sklearn.model_selection import train_test_split
X_train,X_test,y_train,y_test = train_test_split(X,y,test_size =
0.2,random_state=0)
import statsmodels.formula.api as sm
model = sm.OLS(y_train,X_train[:,[0,1,2,3,4,6]])
result = model.fit()
print(result.summary())`
运行此错误:
y_pred = result.predict(X_test)
我的 X_train
的形状是 - (297,7)
我 X_test
的形状是 - (75,7)
dtype
是 numpy.ndarray
之前已经问过这个问题 . 我已经关注了stackoverflow.com上的一些帖子并尝试使用 reshape
函数解决它 . 但是,它没有帮助我 . 任何人都可以解释为什么我收到此错误?什么是解决方案?