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

연습 문제

회귀에서 학습(fit)과 예측(predict)

이제 선형 회귀의 동작 방식을 확인했으니, 미리 로드된 sales_df 데이터셋의 모든 특성(features)을 사용해 다중 선형 회귀 모델을 만들어 보세요. 참고로, 처음 두 행은 다음과 같습니다:

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

그다음 이 모델을 사용해 테스트 특성 값에 기반하여 매출(sales)을 예측합니다.

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

지침

100 XP
  • sales_df의 모든 특성 값을 담은 배열 X와, "sales" 열의 모든 값을 담은 y를 생성하세요.
  • 선형 회귀 모델을 인스턴스화하세요.
  • 학습용 데이터에 모델을 학습(fit)하세요.
  • 테스트 특성으로 sales를 예측하여 y_pred를 생성하세요.