Trénování modelu lineární regrese
Od roku 2015 se každoročně provádí anonymní průzkum mezd mezi evropskými IT specialisty. V roce 2018 se ho dobrovolně zúčastnily stovky respondentů. Data z průzkumu obsahují počet let praxe respondentů a jejich aktuální plat.
Budeš analyzovat vztah mezi těmito dvěma proměnnými a zjistíš, jestli více let praxe odpovídá vyššímu, nebo nižšímu platu.
Tvoje nezávislá proměnná je experience_years a závislá proměnná je current_salary.
Data jsou načtena jako data, spolu s statsmodels.api a pandas jako sm a pd.
Toto cvičení je součástí kurzu
Analýza dat z průzkumů v Pythonu
Pokyny k cvičení
- Definuj proměnné
xay. - Přidej konstantní člen.
- Proveď regresi
OLS()a model.fit(). - Vytiskni souhrnnou tabulku.
Interaktivní cvičení na vyzkoušení si v praxi
Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.
# Define variable, x and y
x = salary_survey.____.____
y = salary_survey.____.____
# Add the constant term
x = ____.____(x)
# Perform .OLS() regression and fit
result = ____.____(y,x).____()
# Print the summary table
print(____.____())