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

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