[ML]mlxtend 모듈로 결정 경계 확인하기

이번 포스팅에서는 mlxtend 모듈에서 제공하는 결정 경계 기능에 대해 알아보도록 하겠습니다.

사이킷런 iris 데이터셋을 사용했습니다.

모듈 설치

pip install mlxtend

모듈 사용

plot_decision_regions 메서드를 사용해서 결정 경계를 시각화할 수 있습니다.

보다 더 자세한 내용은 docs를 참고하겠습니다.

from mlxtend.plotting import plot_decision_regions

plt.figure(figsize=(14, 8))

plot_decision_regions(
    X=iris['data'][:, 2:],
    y=iris['target'],
    clf=iris_tree_clf,
    legend=2
);

결과

모델이 데이터셋을 어떻게 분류했는지 확인할 수 있습니다.