선형 회귀 모델 적합하기
유럽의 IT 전문가들을 대상으로 2015년부터 매년 익명 급여 조사가 진행되었습니다. 2018년에는 수백 명의 응답자가 자발적으로 참여했어요. 설문 데이터에는 응답자의 경력 연수와 현재 연봉이 포함되어 있습니다.
이 두 변수 간의 관계를 분석하여, 경력이 더 많을수록 연봉이 더 높거나 낮아지는지를 확인해 보세요.
독립 변수는 experience_years, 종속 변수는 current_salary입니다.
statsmodels.api와 pandas가 각각 sm, pd로, 데이터는 data로 미리 로드되어 있어요.
이 연습은 강의의 일부입니다
Python으로 설문 데이터 분석하기
연습 안내
- 변수
x와y를 정의하세요. - 상수항을 추가하세요.
OLS()회귀를 수행하고 모델을.fit()하세요.- 요약 표를 출력하세요.
실습형 인터랙티브 연습
이 예제를 이 샘플 코드를 완성하여 풀어보세요.
# 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(____.____())