Блог

Главная - Вопросы по программированию - неправильный dim при использовании .predict с помощью smf.logit

неправильный dim при использовании .predict с помощью smf.logit

#logistic-regression

Вопрос:

У меня есть следующая проблема. Я хотел бы предсказать, используя логистическую регрессию:

 import pandas as pd
import statsmodels.formula.api as smf

model1_logit = smf.logit(
    formula="dep ~   var1   var2", data=model1_data
)

Когда я пытаюсь:

 prediction = model1_logit.predict(pd.DataFrame({'var1': [1], 'var2': [10]}))

или

 prediction = model1_logit.predict(pd.DataFrame({'var1': [1,5,7,9], 'var2': [10,15,20,30]}))

Я получил ошибку ValueError: shapes (6426,3) and (1,2) not aligned: 3 (dim 1) != 1 (dim 0) . Единственный способ, который работает, — это:

 prediction = model1_logit.predict(pd.DataFrame({'var1': [1,3,6], 'var2': [10,15,5]}))

Другими словами, я могу предсказать только три значения. Почему?

Метки: