python - 将 Patsy 用于 Logistic 回归模型的值错误

Question

为了我的问题，我制作了一个非常简单的逻辑模型。这是下面的代码；

import patsy
import pandas as pd
import statsmodels.api as sm

df = pd.DataFrame()

for i in range(5):
    df.at[i, 'response'] = 1
    if i == 3:
        df.at[i,'response'] = 0

df['x'] = range(5)

y, X = patsy.dmatrices('response ~ x', df,return_type = 'dataframe')
logit_model=sm.Logit(y,X)
result=logit_model.fit()

ypred = logit_model.predict(X)
print(ypred)

请原谅我蹩脚的代码，我正在匆忙写这个 - 需要去工作哈哈。这段代码抛出了一个值错误 - ValueError: shape (5,2) and (5,2) not aligned: 2 (dim 1) != 5 (dim 0) about line 18。

我真的不明白这些是如何不对齐的，因为我只是使用 predict() 将训练数据 X 传递回模型。我的感觉是我错过了关于 patsy.dmatrices 的一些东西。

有人有想法吗？

标签： pythonlogistic-regressionpredictpatsy

Answer 1

您将拟合对象分配给result，因此您应该使用它来预测：

result.predict(X)

要获得拟合值，您还可以执行以下操作：

result.fittedvalues