더북(TheBook)

2. 훈련된 회귀 트리 모델을 이용해 테스트 데이터에 대한 예측을 수행합니다.

[in :]

y_pred = regressor.predict(X_test)

3. RMSE로 모델의 성능을 정량화합니다.

[in :]

from sklearn.metrics import mean_squared_error 
from math import sqrt 
sqrt(mean_squared_error(y_test, y_pred))

4. 결과는 다음과 같습니다.

[out:]

4.464255966462036

신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.