Модель с взаимодействием

Промежуточная регрессия со statsmodels в Python

Maarten Van den Broeck

Content Developer at DataCamp

Что такое взаимодействие?

В наборе данных о рыбах

У разных видов рыб разное соотношение массы и длины.
Влияние длины на ожидаемую массу различается для разных видов.

В общем случае

Влияние одной объясняющей переменной на ожидаемый отклик меняется в зависимости от значения другой объясняющей переменной.

Задание взаимодействий

Без взаимодействий

response ~ explntry1 + explntry2

С взаимодействиями (неявными)

response_var ~ explntry1 * explntry2

С взаимодействиями (явными)

response ~ explntry1 + explntry2 + explntry1:explntry2

Без взаимодействий

mass_g ~ length_cm + species

С взаимодействиями (неявными)

mass_g ~ length_cm * species

С взаимодействиями (явными)

mass_g ~ length_cm + species + length_cm:species

Запуск модели

mdl_mass_vs_both = ols("mass_g ~ length_cm * species", data=fish).fit()

print(mdl_mass_vs_both.params)

Intercept                    -1035.3476
species[T.Perch]               416.1725
species[T.Pike]               -505.4767
species[T.Roach]               705.9714
length_cm                       54.5500
length_cm:species[T.Perch]     -15.6385
length_cm:species[T.Pike]       -1.3551
length_cm:species[T.Roach]     -31.2307

Более понятные коэффициенты

mdl_mass_vs_both_inter = ols("mass_g ~ species + species:length_cm + 0", data=fish).fit()

print(mdl_mass_vs_both_inter.params)

species[Bream]             -1035.3476
species[Perch]              -619.1751
species[Pike]              -1540.8243
species[Roach]              -329.3762
species[Bream]:length_cm      54.5500
species[Perch]:length_cm      38.9115
species[Pike]:length_cm       53.1949
species[Roach]:length_cm      23.3193

Знакомые числа

print(mdl_mass_vs_both_inter.params)

species[Bream]             -1035.3476
species[Perch]              -619.1751
species[Pike]              -1540.8243
species[Roach]              -329.3762
species[Bream]:length_cm      54.5500
species[Perch]:length_cm      38.9115
species[Pike]:length_cm       53.1949
species[Roach]:length_cm      23.3193

print(mdl_bream.params)

Intercept   -1035.3476
length_cm      54.5500

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

Промежуточная регрессия со statsmodels в Python