Как построить GLM на Python?

Обобщённые линейные модели в Python

Ita Cirovic Donev

Data Science Consultant

statsmodels

Импорт statsmodels
```
import statsmodels.api as sm
```
Поддержка формул
```
import statsmodels.formula.api as smf
```
Прямое использование glm()
```
from statsmodels.formula.api import glm
```

Процесс построения модели

Описать модель $\rightarrow \texttt{\color{#007AFF}{glm()}}$
Обучить модель $\rightarrow \texttt{\color{#007AFF}{.fit()}}$
Получить сводку $\rightarrow \texttt{\color{#007AFF}{.summary()}}$
Построить прогнозы $\rightarrow \texttt{\color{#007AFF}{.predict()}}$

Описание модели

На основе ФОРМУЛЫ

from statsmodels.formula.api import glm

model = glm(formula, data, family)

На основе МАССИВОВ

import statsmodels.api as sm

X = sm.add_constant(X)
model = sm.glm(y, X, family)

Аргумент formula

$$\texttt{\color{#00A388}{response}} \sim \texttt{\color{#FF6138}{explanatory variable(s)}}$$ $$\texttt{\color{#00A388}{output}} \sim \texttt{\color{#FF6138}{input(s)}}$$

formula = 'y ~ x1 + x2'

$\texttt{\color{#FF6138}{C(x1)}}$ : обработать x1 как категориальную переменную
$\texttt{\color{#FF6138}{-1}}$ : убрать константу (интерсепт)
$\texttt{\color{#FF6138}{x1:x2}}$ : эффект взаимодействия между x1 и x2
$\texttt{\color{#FF6138}{x1*x2}}$ : эффект взаимодействия между x1 и x2 вместе с отдельными переменными
$\texttt{\color{#FF6138}{np.log(x1)}}$ : применить векторизованные функции к переменным модели

Аргумент family

family = sm.families.____()

Семейства распределений:

$\texttt{\color{#007AFF}{Gaussian}(link = sm.families.links.\color{deeppink}{identity()})}$ $\rightarrow$ семейство по умолчанию
$\texttt{\color{#007AFF}{Binomial}(link = sm.families.links.\color{deeppink}{logit()})}$
- $\texttt{\color{deeppink}{probit()}}$, $\texttt{\color{deeppink}{cauchy()}}$, $\texttt{\color{deeppink}{log()}}$, $\texttt{\color{deeppink}{cloglog()}}$
$\texttt{\color{#007AFF}{Poisson}(link = sm.families.links.\color{deeppink}{log()})}$
- $\texttt{\color{deeppink}{identity()}}$ и $\texttt{\color{deeppink}{sqrt()}}$

Другие семейства распределений — на сайте statsmodels.

Сводка модели

print(model_GLM.summary())

                 Generalized Linear Model Regression Results                  
=============================================================================
Dep. Variable:                      y  No. Observations:                  173
Model:                            GLM  Df Residuals:                      171
Model Family:                Binomial  Df Model:                            1
Link Function:                  logit  Scale:                          1.0000
Method:                          IRLS  Log-Likelihood:                -97.226
Date:                Mon, 21 Jan 2019  Deviance:                       194.45
Time:                        11:30:01  Pearson chi2:                     165.
No. Iterations:                     4  Covariance Type:             nonrobust
=============================================================================
                 coef    std err         z      P>|z|      [0.025      0.975]
-----------------------------------------------------------------------------
Intercept    -12.3508      2.629    -4.698      0.000     -17.503      -7.199
width          0.4972      0.102     4.887      0.000       0.298       0.697
=============================================================================

Коэффициенты регрессии

$\texttt{\color{#007AFF}{.params}}$ выводит коэффициенты регрессии

model_GLM.params

Intercept   -12.350818
width         0.497231
dtype: float64

$\texttt{\color{#007AFF}{.conf\_int(alpha=0.05, cols=None)}}$ выводит доверительные интервалы

model_GLM.conf_int()

                   0         1
Intercept -17.503010 -7.198625
width       0.297833  0.696629

Прогнозы

Укажите все переменные модели в тестовых данных
$\texttt{\color{#007AFF}{.predict(\color{#FF931B}{test\_data})}}$ вычисляет прогнозы

model_GLM.predict(test_data)

0    0.029309
1    0.470299
2    0.834983
3    0.972363
4    0.987941

Давайте потренируемся!

Обобщённые линейные модели в Python