Cha 5-1

결정트리

표준화를 하지 않아도 된다

실습코드

from sklearn.tree import DecisionTreeClassifier
import matplotlib.pyplot as plt
from sklearn.tree import plot_tree

dt = DecisionTreeClassifier(random_state=42)
dt.fit(train_scaled, train_target)
print(dt.score(train_scaled, t엔

Untitled

plt.figure(figsize=(10, 7))
plot_tree(dt, max_depth=1, filled=True, feature_names=['alcohol', 'sugar', 'pH'])
# max_depth: 시각화 할 결정트리의 깊이 제한, filled: 매개변수에서 클래스에 맞게 색칠, 
plt.show()

Untitled

지니 불순도

불순도 적용 방법