Как суммировать Logisti c регрессия от statsmodel

Question

Я написал код ниже, но я хотел бы сделать резюме из statsmodel, кто-нибудь может мне помочь, пожалуйста?

Спасибо.

from sklearn.model_selection import train_test_split
from sklearn.linear_model import LogisticRegression

X = df[['age_over_65', 'female_perc', 'foreign_born_perc','bachelors_perc', 'household_income']]
y = df['winner']
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.33, random_state=42)

logmodel = LogisticRegression(solver='lbfgs')
logmodel.fit(X_train,y_train)
model = logmodel.predict(X_test)

Answer 1

Обучение Sci-Kit ориентировано на производительность машинного обучения, а не на статистический вывод.

Если вы хотите увидеть сводные результаты для модели lo git, вам лучше использовать statsmodels

Пример кода ниже.

import statsmodels.formula.api as smf
X = df[['age_over_65', 'female_perc', 'foreign_born_perc','bachelors_perc', 'household_income']]
y = df['winner']
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.33, random_state=42)
XVARS = ['age_over_65', 'female_perc', 'foreign_born_perc','bachelors_perc', 'household_income']
target = 'winner'
model = smf.logit(formula=f"{target} ~ {' + '.join(XVARS)}", data=df.loc[X_train.index])
logmodel = model.fit()
logmodel.summary2()

#to save in a text file.

with open('logit_results.txt'), 'w') as text_file:
    print(logmodel.summary2(), file=text_file)