더북(TheBook)

7.3.5 회귀 알고리즘 비교하기

지금까지 사용한 세 가지 회귀 알고리즘의 성능을 비교해 봅시다.

▼ 표 7-6 회귀 알고리즘의 성능 비교

알고리즘

RMSE

선형 회귀

4.36214129677179

회귀 트리

4.464255966462036

그레이디언트 부스팅 회귀

4.034836373089085

모든 회귀 알고리즘의 성능을 종합해 봤을 때 가장 낮은 RMSE를 기록한 그레이디언트 부스팅 회귀가 성능이 가장 좋습니다. 그 다음으로 성능이 좋은 모델은 선형 회귀입니다. 회귀 트리는 이 문제에서 가장 안 좋은 성능을 기록했습니다.

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