Hayden's Archive

[AI/머신러닝] breast_cancer 데이터를 통해 알아보는 stratify / 가지치기 / 특성 중요도 / 특성 중요도 시각화 / 결정트리 시각화 본문

Study/AI & Math

[AI/머신러닝] breast_cancer 데이터를 통해 알아보는 stratify / 가지치기 / 특성 중요도 / 특성 중요도 시각화 / 결정트리 시각화

_hayden 2020. 7. 23. 17:51

 

stratify
바이너리 데이터일 때 (0과 1로 구성된 데이터일 때)
target이 25%(0에 해당하는 데이터), 75%(1에 해당하는 데이터) 이 비율로 유지한채로 라벨 데이터를 분할할 때 사용

 


정확도 출력

 


가지치기(Pruning)

 


특성 중요도(Feature Importance)

 


특성 중요도 시각화

 


결정트리 시각화