결정 트리(Decision Tree)

bong·2024년 5월 21일

ML

목록 보기
6/8
  • 분류와 회귀에 사용되는 지도 학습 방법
  • 데이터 특성으로부터 추론된 결정 규칙을 통해 값 예측
  • if-then-else 결정 규칙을 통해 데이터 학습
  • 트리 깊이가 깊을 수록 복잡한 모델
from sklearn.metrics import accuracy_score
from sklearn.tree import DecisionTreeClassifier

DT = DecisionTreeClassifier()
DT.fit(x_train, y_train)
pred = DT.predict(x_test)
acc = accuracy_score(pred, y_test)
print(acc)
profile
늅늅

0개의 댓글