[혼공머신] p227 질문이 있습니다.

[오기군]

트리 그림을 출력이 나왔는데요.

import matplotlib.pyplot as plt

from sklearn.tree import plot_tree

plt.figure(figure(10,7))

plot_tree(dt)

plt.show()

여기에서 plot_tree(dt)가 트리를 그리는건데...dt 객체는 단순 dt = DecisionTreeClassifier(random_state=42) 라는건데,

훈련, 테스트 데이터가 어떻게 그려질까요?

dt에는 결정트리를 분류하는 객체일뿐이고, 훈련, 테스트 데이터를 가지고 있지 않는데, plot_tree(dt)로 그림이 그려지는게 이해가 안됩니다.

plot_tree(훈련, 테스트 데이터) 이렇게 들어가야 하는게 아닌지...

[Haesun Park]

안녕하세요. 박해선입니다.

dt 객체에는 fit() 메서드를 통해 훈련된 모델입니다.

이 객체는 훈련 중에 각 노드의 분기에 대한 정보가 포함되어 있습니다.

plot_tree는 이 정보를 바탕으로 그래프를 그립니다.

감사합니다!

2023년 11월 22일 (수) 오후 3:37, 오기군 <kko...@gmail.com>님이 작성:

--
이 메일은 Google 그룹스 '머신러닝/딥러닝 도서 Q&A' 그룹에 가입한 분들에게 전송되는 메시지입니다.
이 그룹에서 탈퇴하고 더 이상 이메일을 받지 않으려면 ml-dl-book-qn...@googlegroups.com에 이메일을 보내세요.
웹에서 이 토론을 보려면 https://groups.google.com/d/msgid/ml-dl-book-qna/7130d704-c1f5-404c-b3f1-e2d4ba179448n%40googlegroups.com을(를) 방문하세요.