파이썬・ML
[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 ); 결과 모델이 데이터셋을 어떻게 분류했는지 확인할 수 있습니다.
