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.034836373089085