인공지능/머신러닝
머신 러닝 : 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) => 적중률을 알 수 있다.