1. Học hỏi
    2. /
  2. Khoa Học
    3. /
  3. Python으로 배우는 트리 기반 Machine Learning
Connected

Bài tập

선형 회귀 vs 회귀 트리

이 연습에서는 dt가 달성한 테스트 세트 RMSE를 선형 회귀 모델과 비교해 보세요. 선형 회귀 모델 lr은 이미 dt와 같은 데이터셋으로 학습해 두었습니다.

특징 행렬 X_test, 레이블 배열 y_test, 학습된 선형 회귀 모델 lr, MSE로 임포트된 mean_squared_error 함수, 그리고 이전 연습에서 구한 rmse_dt가 작업 공간에 준비되어 있습니다.

Hướng dẫn

100 XP

  • 선형 회귀 모델(lr)로 테스트 세트 레이블을 예측하고 결과를 y_pred_lr에 저장하세요.

  • 테스트 세트 MSE를 계산하여 mse_lr에 저장하세요.

  • 테스트 세트 RMSE를 계산하여 rmse_lr에 저장하세요.