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

연습 문제

회귀 모델 학습 및 예측

선형 회귀의 작동 방식을 이해했으니, 이제 미리 로드된 sales_df 데이터셋의 모든 피처를 사용하여 다중 선형 회귀 모델을 만들어 보세요. 데이터셋의 첫 두 행은 다음과 같습니다.

     tv        radio      social_media    sales
1    13000.0   9237.76    2409.57         46677.90
2    41000.0   15886.45   2913.41         150177.83

이 모델을 사용하여 테스트 피처 값을 기반으로 판매량을 예측합니다.

LinearRegression과 train_test_split은 각각의 모듈에서 미리 로드되어 있습니다.

지침

100 XP
  • X의 모든 피처 값을 담은 배열 sales_df와 y 열의 모든 값을 담은 "sales"를 생성하세요.
  • 선형 회귀 모델을 인스턴스화하세요.
  • 학습 데이터에 모델을 피팅하세요.
  • 테스트 피처를 사용하여 판매량(y_pred)을 예측한 결과를 sales에 저장하세요.