문돌이 존버/데이터 분석
2021. 1. 3.
핸즈온 머신러닝 2 복습하기(챕터 6: 결정 트리)
CH 6. 결정 트리 결정 트리(decision tree)는 분류와 회귀 작업 그리고 다중출력 작업도 가능한 다재다능한 머신러닝 알고리즘이다. 매우 복잡한 데이터셋도 학습할 수 있는 강력한 알고리즘이다. 6.1 결정 트리 학습과 시각화 from sklearn.datasets import load_iris from sklearn.tree import DecisionTreeClassifier iris = load_iris() X = iris.data[:, 2:] # 꽃잎 길이와 너비 y = iris.target tree_clf = DecisionTreeClassifier(max_depth=2) tree_clf.fit(X, y) from sklearn.tree import export_graphviz expo..
