Jeden model z interakcją

Regresja średnio zaawansowana ze statsmodels w Pythonie

Maarten Van den Broeck

Content Developer at DataCamp

Czym jest interakcja?

W zbiorze danych o rybach

Różne gatunki ryb mają różny stosunek masy do długości.
Wpływ długości na oczekiwaną masę różni się w zależności od gatunku.

Ogólnie

Wpływ jednej zmiennej objaśniającej na oczekiwaną odpowiedź zależy od wartości innej zmiennej objaśniającej.

Określanie interakcji

Bez interakcji

response ~ explntry1 + explntry2

Z interakcjami (niejawnie)

response_var ~ explntry1 * explntry2

Z interakcjami (jawnie)

response ~ explntry1 + explntry2 + explntry1:explntry2

Bez interakcji

mass_g ~ length_cm + species

Z interakcjami (niejawnie)

mass_g ~ length_cm * species

Z interakcjami (jawnie)

mass_g ~ length_cm + species + length_cm:species

Uruchamianie modelu

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

Łatwiejsze do interpretacji współczynniki

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

Znajome liczby

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

Czas na ćwiczenia!

Regresja średnio zaawansowana ze statsmodels w Pythonie