기억 저장소

클라우드 기반 인공지능 개발과 DevOps 실무

인공지능/머신러닝

머신 러닝 : Decision Tree

하늘.98 2021. 11. 25. 18:01

결정 트리(Decision Tree, 의사결정트리, 의사결정나무라고도 함)는 분류(Classification)와 회귀(Regression) 모두 가능한 지도 학습 모델 중 하나입니다. 결정 트리는 스무고개 하듯이 예/아니오 질문을 이어가며 학습합니다.

read_csv로 파일을 불러온 뒤 필요한 X , y 값을 분리해준다.

X (학습시킬 값)  ,   y (학습시킨 값의 답) 

트레이닝과 테스트 값을 나눠준다.  

from sklearn.tree import DecisionTreeClassifier 

라이브러리를 불러온 뒤 위에 그림과 같이 대입하여 사용해주면 된다 . 

confusion_matrix(y_test,y_pred)              => 맞춘값과 틀린 값을 알 수 있다.

accuracy_score(y_test,y_pred)               => 적중률을 알 수 있다.