1. 학습
    2. /
  2. 강의
    3. /
  3. scikit-learn으로 배우는 지도 학습
Connected

연습 문제

선형 회귀 모델 구축하기

특성 배열과 타깃 배열을 생성했으니, 이제 모든 특성 값과 타깃 값을 사용하여 선형 회귀 모델을 학습시켜 보겠습니다.

특성 값과 타깃 값 간의 관계를 파악하는 것이 목표이므로, 데이터를 학습 세트와 테스트 세트로 분리할 필요는 없습니다.

X와 y는 다음과 같이 미리 로드되어 있습니다:

y = sales_df["sales"].values
X = sales_df["radio"].values.reshape(-1, 1)

지침

100 XP
  • LinearRegression을 임포트하세요.
  • 선형 회귀 모델을 인스턴스화하세요.
  • X를 사용하여 판매량 예측값을 구하고, predictions에 저장하세요.