En modell med en interaktion

Intermediär regression med statsmodels i Python

Maarten Van den Broeck

Content Developer at DataCamp

Vad är en interaktion?

I fiskedatasetet

Olika fiskarter har olika förhållande mellan massa och längd.
Längdens effekt på den förväntade massan varierar mellan arter.

Mer generellt

Effekten av en förklaringsvariabel på det förväntade utfallet beror på värdet hos en annan förklaringsvariabel.

Ange interaktioner

Utan interaktioner

response ~ explntry1 + explntry2

Med interaktioner (implicit)

response_var ~ explntry1 * explntry2

Med interaktioner (explicit)

response ~ explntry1 + explntry2 + explntry1:explntry2

Utan interaktioner

mass_g ~ length_cm + species

Med interaktioner (implicit)

mass_g ~ length_cm * species

Med interaktioner (explicit)

mass_g ~ length_cm + species + length_cm:species

Köra modellen

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

Lättare att förstå koefficienter

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

Bekanta tal

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

Nu kör vi en övning!

Intermediär regression med statsmodels i Python