Одна модель з взаємодією

Середній рівень регресії зі 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